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
wFaceExtractorNet.hpp
Go to the documentation of this file.
1 #ifndef OPENPOSE_FACE_W_FACE_DETECTOR_NET_HPP
2 #define OPENPOSE_FACE_W_FACE_DETECTOR_NET_HPP
3 
7 
8 namespace op
9 {
10  template<typename TDatums>
11  class WFaceExtractorNet : public Worker<TDatums>
12  {
13  public:
14  explicit WFaceExtractorNet(const std::shared_ptr<FaceExtractorNet>& faceExtractorNet);
15 
17 
18  void work(TDatums& tDatums);
19 
20  private:
21  std::shared_ptr<FaceExtractorNet> spFaceExtractorNet;
22 
23  DELETE_COPY(WFaceExtractorNet);
24  };
25 }
26 
27 
28 
29 
30 
31 // Implementation
33 namespace op
34 {
35  template<typename TDatums>
36  WFaceExtractorNet<TDatums>::WFaceExtractorNet(const std::shared_ptr<FaceExtractorNet>& faceExtractorNet) :
37  spFaceExtractorNet{faceExtractorNet}
38  {
39  }
40 
41  template<typename TDatums>
43  {
44  spFaceExtractorNet->initializationOnThread();
45  }
46 
47  template<typename TDatums>
48  void WFaceExtractorNet<TDatums>::work(TDatums& tDatums)
49  {
50  try
51  {
52  if (checkNoNullNorEmpty(tDatums))
53  {
54  // Debugging log
55  dLog("", Priority::Low, __LINE__, __FUNCTION__, __FILE__);
56  // Profiling speed
57  const auto profilerKey = Profiler::timerInit(__LINE__, __FUNCTION__, __FILE__);
58  // Extract people face
59  for (auto& tDatum : *tDatums)
60  {
61  spFaceExtractorNet->forwardPass(tDatum.faceRectangles, tDatum.cvInputData);
62  tDatum.faceHeatMaps = spFaceExtractorNet->getHeatMaps().clone();
63  tDatum.faceKeypoints = spFaceExtractorNet->getFaceKeypoints().clone();
64  }
65  // Profiling speed
66  Profiler::timerEnd(profilerKey);
67  Profiler::printAveragedTimeMsOnIterationX(profilerKey, __LINE__, __FUNCTION__, __FILE__);
68  // Debugging log
69  dLog("", Priority::Low, __LINE__, __FUNCTION__, __FILE__);
70  }
71  }
72  catch (const std::exception& e)
73  {
74  this->stop();
75  tDatums = nullptr;
76  error(e.what(), __LINE__, __FUNCTION__, __FILE__);
77  }
78  }
79 
81 }
82 
83 #endif // OPENPOSE_FACE_W_FACE_DETECTOR_NET_HPP
Definition: worker.hpp:9
void work(TDatums &tDatums)
Definition: wFaceExtractorNet.hpp:48
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="")
Definition: wFaceExtractorNet.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
bool checkNoNullNorEmpty(const TPointerContainer &tPointerContainer)
Definition: pointerContainer.hpp:7
void initializationOnThread()
Definition: wFaceExtractorNet.hpp:42
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)
COMPILE_TEMPLATE_DATUM(WPoseTriangulation)
WFaceExtractorNet(const std::shared_ptr< FaceExtractorNet > &faceExtractorNet)
Definition: wFaceExtractorNet.hpp:36
static void timerEnd(const std::string &key)