| | #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 |
| |
|