1 #ifndef OPENPOSE_NET_NMS_CAFFE_HPP
2 #define OPENPOSE_NET_NMS_CAFFE_HPP
22 const int maxPeaks,
const int outputChannels = -1,
const int gpuID = 0);
24 virtual inline const char*
type()
const {
return "Nms"; }
53 std::unique_ptr<ImplNmsCaffe> upImpl;
61 #endif // OPENPOSE_NET_NMS_CAFFE_HPP
#define DELETE_COPY(className)
Definition: macros.hpp:34
virtual void LayerSetUp(const std::vector< caffe::Blob< T > * > &bottom, const std::vector< caffe::Blob< T > * > &top)
virtual void Forward(const std::vector< caffe::Blob< T > * > &bottom, const std::vector< caffe::Blob< T > * > &top)
virtual void Forward_cpu(const std::vector< caffe::Blob< T > * > &bottom, const std::vector< caffe::Blob< T > * > &top)
void setOffset(const Point< T > &offset)
virtual const char * type() const
Definition: nmsCaffe.hpp:24
virtual void Forward_ocl(const std::vector< caffe::Blob< T > * > &bottom, const std::vector< caffe::Blob< T > * > &top)
virtual void Forward_gpu(const std::vector< caffe::Blob< T > * > &bottom, const std::vector< caffe::Blob< T > * > &top)
virtual void Reshape(const std::vector< caffe::Blob< T > * > &bottom, const std::vector< caffe::Blob< T > * > &top, const int maxPeaks, const int outputChannels=-1, const int gpuID=0)
Definition: macros.hpp:81
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 Backward_gpu(const std::vector< caffe::Blob< T > * > &top, const std::vector< bool > &propagate_down, const std::vector< caffe::Blob< T > * > &bottom)
void setThreshold(const T threshold)
std::vector< T, Alloc > vector
Definition: cl2.hpp:567
Definition: nmsCaffe.hpp:12