OpenPose  1.0.0rc2
OpenPose: A Real-Time Multi-Person Key-Point Detection And Multi-Threading C++ Library
 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Macros
wPoseExtractor.hpp
Go to the documentation of this file.
1 #ifndef OPENPOSE_POSE_W_POSE_EXTRACTOR_HPP
2 #define OPENPOSE_POSE_W_POSE_EXTRACTOR_HPP
3 
7 
8 namespace op
9 {
10  template<typename TDatums>
11  class WPoseExtractor : public Worker<TDatums>
12  {
13  public:
14  explicit WPoseExtractor(const std::shared_ptr<PoseExtractor>& poseExtractorSharedPtr);
15 
16  virtual ~WPoseExtractor();
17 
19 
20  void work(TDatums& tDatums);
21 
22  private:
23  std::shared_ptr<PoseExtractor> spPoseExtractor;
24 
25  DELETE_COPY(WPoseExtractor);
26  };
27 }
28 
29 
30 
31 
32 
33 // Implementation
35 namespace op
36 {
37  template<typename TDatums>
38  WPoseExtractor<TDatums>::WPoseExtractor(const std::shared_ptr<PoseExtractor>& poseExtractorSharedPtr) :
39  spPoseExtractor{poseExtractorSharedPtr}
40  {
41  }
42 
43  template<typename TDatums>
45  {
46  }
47 
48  template<typename TDatums>
50  {
51  try
52  {
53  spPoseExtractor->initializationOnThread();
54  }
55  catch (const std::exception& e)
56  {
57  error(e.what(), __LINE__, __FUNCTION__, __FILE__);
58  }
59  }
60 
61  template<typename TDatums>
62  void WPoseExtractor<TDatums>::work(TDatums& tDatums)
63  {
64  try
65  {
66  if (checkNoNullNorEmpty(tDatums))
67  {
68  // Debugging log
69  dLog("", Priority::Low, __LINE__, __FUNCTION__, __FILE__);
70  // Profiling speed
71  const auto profilerKey = Profiler::timerInit(__LINE__, __FUNCTION__, __FILE__);
72  // Extract people pose
73  for (auto i = 0u ; i < tDatums->size() ; i++)
74  // for (auto& tDatum : *tDatums)
75  {
76  auto& tDatum = (*tDatums)[i];
77  // OpenPose net forward pass
78  spPoseExtractor->forwardPass(tDatum.inputNetData,
79  Point<int>{tDatum.cvInputData.cols, tDatum.cvInputData.rows},
80  tDatum.scaleInputToNetInputs, tDatum.id);
81  // OpenPose keypoint detector
82  tDatum.poseCandidates = spPoseExtractor->getCandidatesCopy();
83  tDatum.poseHeatMaps = spPoseExtractor->getHeatMapsCopy();
84  tDatum.poseKeypoints = spPoseExtractor->getPoseKeypoints().clone();
85  tDatum.poseScores = spPoseExtractor->getPoseScores().clone();
86  tDatum.scaleNetToOutput = spPoseExtractor->getScaleNetToOutput();
87  // Keep desired top N people
88  spPoseExtractor->keepTopPeople(tDatum.poseKeypoints, tDatum.poseScores);
89  // ID extractor (experimental)
90  tDatum.poseIds = spPoseExtractor->extractIdsLockThread(tDatum.poseKeypoints, tDatum.cvInputData,
91  i, tDatum.id);
92  // Tracking (experimental)
93  spPoseExtractor->trackLockThread(tDatum.poseKeypoints, tDatum.poseIds, tDatum.cvInputData, i,
94  tDatum.id);
95  }
96  // Profiling speed
97  Profiler::timerEnd(profilerKey);
98  Profiler::printAveragedTimeMsOnIterationX(profilerKey, __LINE__, __FUNCTION__, __FILE__);
99  // Debugging log
100  dLog("", Priority::Low, __LINE__, __FUNCTION__, __FILE__);
101  }
102  }
103  catch (const std::exception& e)
104  {
105  this->stop();
106  tDatums = nullptr;
107  error(e.what(), __LINE__, __FUNCTION__, __FILE__);
108  }
109  }
110 
111  COMPILE_TEMPLATE_DATUM(WPoseExtractor);
112 }
113 
114 #endif // OPENPOSE_POSE_W_POSE_EXTRACTOR_HPP
WPoseExtractor(const std::shared_ptr< PoseExtractor > &poseExtractorSharedPtr)
Definition: wPoseExtractor.hpp:38
Definition: worker.hpp:9
#define COMPILE_TEMPLATE_DATUM(templateName)
Definition: datum.hpp:390
static const std::string timerInit(const int line, const std::string &function, const std::string &file)
OP_API void error(const std::string &message, const int line=-1, const std::string &function="", const std::string &file="")
virtual ~WPoseExtractor()
Definition: wPoseExtractor.hpp:44
void initializationOnThread()
Definition: wPoseExtractor.hpp:49
Definition: wPoseExtractor.hpp:11
void dLog(const T &message, const Priority priority=Priority::Max, const int line=-1, const std::string &function="", const std::string &file="")
Definition: errorAndLog.hpp:53
void work(TDatums &tDatums)
Definition: wPoseExtractor.hpp:62
bool checkNoNullNorEmpty(const TPointerContainer &tPointerContainer)
Definition: pointerContainer.hpp:7
static void printAveragedTimeMsOnIterationX(const std::string &key, const int line, const std::string &function, const std::string &file, const unsigned long long x=DEFAULT_X)
static void timerEnd(const std::string &key)