| #ifndef OPENPOSE_NET_NET_OPEN_CV_HPP |
| #define OPENPOSE_NET_NET_OPEN_CV_HPP |
|
|
| #include <openpose/core/common.hpp> |
| #include <openpose/net/net.hpp> |
|
|
| namespace op |
| { |
| class OP_API NetOpenCv : public Net |
| { |
| public: |
| NetOpenCv(const std::string& caffeProto, const std::string& caffeTrainedModel, const int gpuId = 0); |
|
|
| virtual ~NetOpenCv(); |
|
|
| void initializationOnThread(); |
|
|
| void forwardPass(const Array<float>& inputNetData) const; |
|
|
| std::shared_ptr<ArrayCpuGpu<float>> getOutputBlobArray() const; |
|
|
| private: |
| |
| |
| struct ImplNetOpenCv; |
| std::unique_ptr<ImplNetOpenCv> upImpl; |
|
|
| |
| |
| DELETE_COPY(NetOpenCv); |
| }; |
| } |
|
|
| #endif |
|
|