OpenPose  1.0.0rc2
OpenPose: A Real-Time Multi-Person Key-Point Detection And Multi-Threading C++ Library
 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Macros
resizeAndMergeCaffe.hpp
Go to the documentation of this file.
1 #ifndef OPENPOSE_NET_RESIZE_AND_MERGE_CAFFE_HPP
2 #define OPENPOSE_NET_RESIZE_AND_MERGE_CAFFE_HPP
3 
5 
6 namespace op
7 {
8  // It mostly follows the Caffe::layer implementation, so Caffe users can easily use it. However, in order to keep
9  // the compatibility with any generic Caffe version, we keep this 'layer' inside our library rather than in the
10  // Caffe code.
11  template <typename T>
13  {
14  public:
15  explicit ResizeAndMergeCaffe();
16 
17  virtual ~ResizeAndMergeCaffe();
18 
19  virtual void LayerSetUp(const std::vector<caffe::Blob<T>*>& bottom, const std::vector<caffe::Blob<T>*>& top);
20 
21  virtual void Reshape(const std::vector<caffe::Blob<T>*>& bottom, const std::vector<caffe::Blob<T>*>& top,
22  const T netFactor, const T scaleFactor, const bool mergeFirstDimension = true,
23  const int gpuID = 0);
24 
25  virtual inline const char* type() const { return "ResizeAndMerge"; }
26 
27  void setScaleRatios(const std::vector<T>& scaleRatios);
28 
29  virtual void Forward_cpu(const std::vector<caffe::Blob<T>*>& bottom, const std::vector<caffe::Blob<T>*>& top);
30 
31  virtual void Forward_gpu(const std::vector<caffe::Blob<T>*>& bottom, const std::vector<caffe::Blob<T>*>& top);
32 
33  virtual void Forward_ocl(const std::vector<caffe::Blob<T>*>& bottom, const std::vector<caffe::Blob<T>*>& top);
34 
35  virtual void Backward_cpu(const std::vector<caffe::Blob<T>*>& top, const std::vector<bool>& propagate_down,
36  const std::vector<caffe::Blob<T>*>& bottom);
37 
38  virtual void Backward_gpu(const std::vector<caffe::Blob<T>*>& top, const std::vector<bool>& propagate_down,
39  const std::vector<caffe::Blob<T>*>& bottom);
40 
41  private:
42  std::vector<T*> mTempGPUData;
43  std::vector<T> mScaleRatios;
44  std::vector<std::array<int, 4>> mBottomSizes;
45  std::array<int, 4> mTopSize;
46  int mGpuID;
47 
48  DELETE_COPY(ResizeAndMergeCaffe);
49  };
50 }
51 
52 #endif // OPENPOSE_NET_RESIZE_AND_MERGE_CAFFE_HPP
void setScaleRatios(const std::vector< T > &scaleRatios)
virtual void Backward_gpu(const std::vector< caffe::Blob< T > * > &top, const std::vector< bool > &propagate_down, const std::vector< caffe::Blob< T > * > &bottom)
virtual ~ResizeAndMergeCaffe()
virtual void Forward_cpu(const std::vector< caffe::Blob< T > * > &bottom, const std::vector< caffe::Blob< T > * > &top)
Definition: resizeAndMergeCaffe.hpp:12
virtual void Reshape(const std::vector< caffe::Blob< T > * > &bottom, const std::vector< caffe::Blob< T > * > &top, const T netFactor, const T scaleFactor, const bool mergeFirstDimension=true, const int gpuID=0)
virtual const char * type() const
Definition: resizeAndMergeCaffe.hpp:25
virtual void Forward_ocl(const std::vector< caffe::Blob< T > * > &bottom, const std::vector< caffe::Blob< T > * > &top)
Definition: macros.hpp:80
virtual void Backward_cpu(const std::vector< caffe::Blob< T > * > &top, const std::vector< bool > &propagate_down, const std::vector< caffe::Blob< T > * > &bottom)
virtual void LayerSetUp(const std::vector< caffe::Blob< T > * > &bottom, const std::vector< caffe::Blob< T > * > &top)
std::vector< T, Alloc > vector
Definition: cl2.hpp:567
virtual void Forward_gpu(const std::vector< caffe::Blob< T > * > &bottom, const std::vector< caffe::Blob< T > * > &top)