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
nmsCaffe.hpp
Go to the documentation of this file.
1 #ifndef OPENPOSE_NET_NMS_CAFFE_HPP
2 #define OPENPOSE_NET_NMS_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 NmsCaffe();
16 
17  virtual ~NmsCaffe();
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 int maxPeaks, const int outputChannels = -1, const int gpuID = 0);
23 
24  virtual inline const char* type() const { return "Nms"; }
25 
26  void setThreshold(const T threshold);
27 
28  // Empirically gives better results (copied from Matlab original code)
29  void setOffset(const Point<T>& offset);
30 
31  virtual void Forward_cpu(const std::vector<caffe::Blob<T>*>& bottom, const std::vector<caffe::Blob<T>*>& top);
32 
33  virtual void Forward_gpu(const std::vector<caffe::Blob<T>*>& bottom, const std::vector<caffe::Blob<T>*>& top);
34 
35  virtual void Forward_ocl(const std::vector<caffe::Blob<T>*>& bottom, const std::vector<caffe::Blob<T>*>& top);
36 
37  virtual void Backward_cpu(const std::vector<caffe::Blob<T>*>& top, const std::vector<bool>& propagate_down,
38  const std::vector<caffe::Blob<T>*>& bottom);
39 
40  virtual void Backward_gpu(const std::vector<caffe::Blob<T>*>& top, const std::vector<bool>& propagate_down,
41  const std::vector<caffe::Blob<T>*>& bottom);
42 
43  private:
44  T mThreshold;
45  Point<T> mOffset;
46  int mGpuID;
47 
48  // PIMPL idiom
49  // http://www.cppsamples.com/common-tasks/pimpl.html
50  struct ImplNmsCaffe;
51  std::unique_ptr<ImplNmsCaffe> upImpl;
52 
53  // PIMP requires DELETE_COPY & destructor, or extra code
54  // http://oliora.github.io/2015/12/29/pimpl-and-rule-of-zero.html
56  };
57 }
58 
59 #endif // OPENPOSE_NET_NMS_CAFFE_HPP
#define DELETE_COPY(className)
Definition: macros.hpp:33
Definition: point.hpp:10
virtual const char * type() const
Definition: nmsCaffe.hpp:24
Definition: macros.hpp:80
std::vector< T, Alloc > vector
Definition: cl2.hpp:567
Definition: nmsCaffe.hpp:12
#define OP_API
Definition: macros.hpp:18