intermediate_structures.h

Go to the documentation of this file.

 
#ifndef SRC_INTERMEDIATE_STRUCTURES_H_
#define SRC_INTERMEDIATE_STRUCTURES_H_
 
#include <array>
#include <string>
 
enum BodyPart {
  head_top,
  BodyPartMin = head_top,
  upper_neck,
  right_shoulder,
  right_elbow,
  right_wrist,
  thorax,
  left_shoulder,
  left_elbow,
  left_wrist,
  pelvis,
  right_hip,
  right_knee,
  right_ankle,
  left_hip,
  left_knee,
  left_ankle,
  BodyPartMax = left_ankle
};
 
enum Joint {
  Head = 0,
  JointMin = Head,
  Neck = 1,
  Shoulder = 2,
  Hip = 3,
  Knee = 4,
  Foot = 5,
  JointMax = Foot
};
 
namespace PreProcessing {
struct PreProcessedImage {
  float* image;
};
}  // namespace PreProcessing
 
namespace PostProcessing {
 
enum Status { Untrustworthy, Trustworthy };
 
std::string stringStatus(Status s);
 
struct Coordinate {
  float x;        
  float y;        
  Status status;  
};
 
struct ProcessedResults {
  std::array<Coordinate, JointMax + 1> body_parts;
};
}  // namespace PostProcessing
 
namespace Inference {
 
struct Result {
  float confidence;
  size_t x;  
  size_t y;  
};
 
struct Coordinate {
  float x;  
  float y;  
  float confidence;
};
 
struct InferenceResults {
  std::array<Coordinate, BodyPartMax + 1> body_parts;
};
}  // namespace Inference
 
#endif  // SRC_INTERMEDIATE_STRUCTURES_H_