File size: 2,578 Bytes
3bbb319 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 |
#ifndef OPENPOSE_EXAMPLES_TUTORIAL_W_USER_POST_PROCESSING_HPP
#define OPENPOSE_EXAMPLES_TUTORIAL_W_USER_POST_PROCESSING_HPP
#include <openpose/core/common.hpp>
#include <openpose/core/cvMatToOpInput.hpp>
#include <openpose/thread/worker.hpp>
#include "userPostProcessing.hpp"
namespace op
{
template<typename TDatums>
class WUserPostProcessing : public Worker<TDatums>
{
public:
WUserPostProcessing(const std::shared_ptr<UserPostProcessing>& userPostProcessing);
void initializationOnThread();
void work(TDatums& tDatums);
private:
std::shared_ptr<UserPostProcessing> spUserPostProcessing;
DELETE_COPY(WUserPostProcessing);
};
}
// Implementation
#include <openpose/utilities/pointerContainer.hpp>
namespace op
{
template<typename TDatums>
WUserPostProcessing<TDatums>::WUserPostProcessing(const std::shared_ptr<UserPostProcessing>& userPostProcessing) :
spUserPostProcessing{userPostProcessing}
{
}
template<typename TDatums>
void WUserPostProcessing<TDatums>::initializationOnThread()
{
}
template<typename TDatums>
void WUserPostProcessing<TDatums>::work(TDatums& tDatums)
{
try
{
if (checkNoNullNorEmpty(tDatums))
{
// Debugging log
opLogIfDebug("", Priority::Low, __LINE__, __FUNCTION__, __FILE__);
// Profiling speed
const auto profilerKey = Profiler::timerInit(__LINE__, __FUNCTION__, __FILE__);
for (auto& datum : *tDatums)
{
// THESE 2 ARE THE ONLY LINES THAT THE USER MUST MODIFY ON THIS HPP FILE, by using the proper
// function and datum elements
cv::Mat cvOutputData = OP_OP2CVMAT(datum->cvOutputData);
spUserPostProcessing->doSomething(cvOutputData, cvOutputData);
}
// Profiling speed
Profiler::timerEnd(profilerKey);
Profiler::printAveragedTimeMsOnIterationX(profilerKey, __LINE__, __FUNCTION__, __FILE__);
// Debugging log
opLogIfDebug("", Priority::Low, __LINE__, __FUNCTION__, __FILE__);
}
}
catch (const std::exception& e)
{
this->stop();
tDatums = nullptr;
error(e.what(), __LINE__, __FUNCTION__, __FILE__);
}
}
// COMPILE_TEMPLATE_DATUM(WUserPostProcessing);
}
#endif // OPENPOSE_EXAMPLES_TUTORIAL_W_USER_POST_PROCESSING_HPP
|