1 #ifndef OPENPOSE_CORE_NMS_CAFFE_HPP
2 #define OPENPOSE_CORE_NMS_CAFFE_HPP
22 const int maxPeaks,
const int outputChannels = -1);
24 virtual inline const char*
type()
const {
return "Nms"; }
26 void setThreshold(
const T threshold);
32 virtual void Backward_cpu(
const std::vector<
caffe::Blob<T>*>& top,
const std::vector<bool>& propagate_down,
35 virtual void Backward_gpu(
const std::vector<
caffe::Blob<T>*>& top,
const std::vector<bool>& propagate_down,
44 std::unique_ptr<ImplNmsCaffe> upImpl;
52 #endif // OPENPOSE_CORE_NMS_CAFFE_HPP
#define DELETE_COPY(className)
Definition: macros.hpp:24
virtual const char * type() const
Definition: nmsCaffe.hpp:24
Definition: macros.hpp:71
Definition: nmsCaffe.hpp:12
#define OP_API
Definition: macros.hpp:9