OpenPose 1.0.0rc2 OpenPose: A Real-Time Multi-Person Key-Point Detection And Multi-Threading C++ Library

- Main Page
- Namespaces
- Classes
- Files

- File List
- File Members

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
4 #include <openpose/core/common.hpp>
5 #include <openpose/pose/poseExtractor.hpp>
6 #include <openpose/thread/worker.hpp>
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
18 void initializationOnThread();
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
34 #include <openpose/utilities/pointerContainer.hpp>
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>
44 WPoseExtractor<TDatums>::~WPoseExtractor()
45 {
46 }
47
48 template<typename TDatums>
49 void WPoseExtractor<TDatums>::initializationOnThread()
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(
91 tDatum.poseKeypoints, tDatum.cvInputData, i, tDatum.id);
92 // Tracking (experimental)
93 spPoseExtractor->trackLockThread(
94 tDatum.poseKeypoints, tDatum.poseIds, tDatum.cvInputData, i, 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
op::Priority::Low
op::WPoseExtractor::WPoseExtractor
WPoseExtractor(const std::shared_ptr< PoseExtractor > &poseExtractorSharedPtr)
Definition: wPoseExtractor.hpp:38
op::Worker
Definition: worker.hpp:9
COMPILE_TEMPLATE_DATUM
#define COMPILE_TEMPLATE_DATUM(templateName)
Definition: datum.hpp:395
op::Point< int >
common.hpp
op::Profiler::timerInit
static const std::string timerInit(const int line, const std::string &function, const std::string &file)
op::error
OP_API void error(const std::string &message, const int line=-1, const std::string &function="", const std::string &file="")
op::WPoseExtractor::~WPoseExtractor
virtual ~WPoseExtractor()
Definition: wPoseExtractor.hpp:44
pointerContainer.hpp
op::WPoseExtractor::initializationOnThread
void initializationOnThread()
Definition: wPoseExtractor.hpp:49
op::WPoseExtractor
Definition: wPoseExtractor.hpp:11
op::dLog
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
worker.hpp
op::WPoseExtractor::work
void work(TDatums &tDatums)
Definition: wPoseExtractor.hpp:62
op::checkNoNullNorEmpty
bool checkNoNullNorEmpty(const TPointerContainer &tPointerContainer)
Definition: pointerContainer.hpp:7
op::Profiler::printAveragedTimeMsOnIterationX
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)
op::Profiler::timerEnd
static void timerEnd(const std::string &key)
poseExtractor.hpp