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
wScaleAndSizeExtractor.hpp
Go to the documentation of this file.
1 #ifndef OPENPOSE_CORE_W_SCALE_AND_SIZE_EXTRACTOR_HPP
2 #define OPENPOSE_CORE_W_SCALE_AND_SIZE_EXTRACTOR_HPP
3 
7 
8 namespace op
9 {
10  template<typename TDatums>
11  class WScaleAndSizeExtractor : public Worker<TDatums>
12  {
13  public:
14  explicit WScaleAndSizeExtractor(const std::shared_ptr<ScaleAndSizeExtractor>& scaleAndSizeExtractor);
15 
16  virtual ~WScaleAndSizeExtractor();
17 
19 
20  void work(TDatums& tDatums);
21 
22  private:
23  const std::shared_ptr<ScaleAndSizeExtractor> spScaleAndSizeExtractor;
24 
25  DELETE_COPY(WScaleAndSizeExtractor);
26  };
27 }
28 
29 
30 
31 
32 
33 // Implementation
35 namespace op
36 {
37  template<typename TDatums>
39  const std::shared_ptr<ScaleAndSizeExtractor>& scaleAndSizeExtractor) :
40  spScaleAndSizeExtractor{scaleAndSizeExtractor}
41  {
42  }
43 
44  template<typename TDatums>
46  {
47  }
48 
49  template<typename TDatums>
51  {
52  }
53 
54  template<typename TDatums>
56  {
57  try
58  {
59  if (checkNoNullNorEmpty(tDatums))
60  {
61  // Debugging log
62  dLog("", Priority::Low, __LINE__, __FUNCTION__, __FILE__);
63  // Profiling speed
64  const auto profilerKey = Profiler::timerInit(__LINE__, __FUNCTION__, __FILE__);
65  // cv::Mat -> float*
66  for (auto& tDatum : *tDatums)
67  {
68  const Point<int> inputSize{tDatum.cvInputData.cols, tDatum.cvInputData.rows};
69  std::tie(tDatum.scaleInputToNetInputs, tDatum.netInputSizes, tDatum.scaleInputToOutput,
70  tDatum.netOutputSize) = spScaleAndSizeExtractor->extract(inputSize);
71  }
72  // Profiling speed
73  Profiler::timerEnd(profilerKey);
74  Profiler::printAveragedTimeMsOnIterationX(profilerKey, __LINE__, __FUNCTION__, __FILE__);
75  // Debugging log
76  dLog("", Priority::Low, __LINE__, __FUNCTION__, __FILE__);
77  }
78  }
79  catch (const std::exception& e)
80  {
81  this->stop();
82  tDatums = nullptr;
83  error(e.what(), __LINE__, __FUNCTION__, __FILE__);
84  }
85  }
86 
88 }
89 
90 #endif // OPENPOSE_CORE_W_SCALE_AND_SIZE_EXTRACTOR_HPP
Definition: worker.hpp:9
Definition: wScaleAndSizeExtractor.hpp:11
virtual ~WScaleAndSizeExtractor()
Definition: wScaleAndSizeExtractor.hpp:45
void initializationOnThread()
Definition: wScaleAndSizeExtractor.hpp:50
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="")
WScaleAndSizeExtractor(const std::shared_ptr< ScaleAndSizeExtractor > &scaleAndSizeExtractor)
Definition: wScaleAndSizeExtractor.hpp:38
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 work(TDatums &tDatums)
Definition: wScaleAndSizeExtractor.hpp:55
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)
static void timerEnd(const std::string &key)