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