1 #ifndef OPENPOSE_POSE_W_POSE_EXTRACTOR_HPP
2 #define OPENPOSE_POSE_W_POSE_EXTRACTOR_HPP
10 template<
typename TDatums>
14 explicit WPoseExtractor(
const std::shared_ptr<PoseExtractor>& poseExtractorSharedPtr);
20 void work(TDatums& tDatums);
23 std::shared_ptr<PoseExtractor> spPoseExtractor;
37 template<
typename TDatums>
39 spPoseExtractor{poseExtractorSharedPtr}
43 template<
typename TDatums>
48 template<
typename TDatums>
53 spPoseExtractor->initializationOnThread();
55 catch (
const std::exception& e)
57 error(e.what(), __LINE__, __FUNCTION__, __FILE__);
61 template<
typename TDatums>
73 for (
auto i = 0u ; i < tDatums->size() ; i++)
76 auto& tDatum = (*tDatums)[i];
78 spPoseExtractor->forwardPass(tDatum.inputNetData,
79 Point<int>{tDatum.cvInputData.cols, tDatum.cvInputData.rows},
80 tDatum.scaleInputToNetInputs, tDatum.id);
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();
88 spPoseExtractor->keepTopPeople(tDatum.poseKeypoints, tDatum.poseScores);
90 tDatum.poseIds = spPoseExtractor->extractIdsLockThread(
91 tDatum.poseKeypoints, tDatum.cvInputData, i, tDatum.id);
93 spPoseExtractor->trackLockThread(
94 tDatum.poseKeypoints, tDatum.poseIds, tDatum.cvInputData, i, tDatum.id);
103 catch (
const std::exception& e)
107 error(e.what(), __LINE__, __FUNCTION__, __FILE__);
114 #endif // OPENPOSE_POSE_W_POSE_EXTRACTOR_HPP
#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="")
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
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)