|
| NmsCaffe () |
|
virtual | ~NmsCaffe () |
|
virtual void | LayerSetUp (const std::vector< ArrayCpuGpu< T > * > &bottom, const std::vector< ArrayCpuGpu< T > * > &top) |
|
virtual void | Reshape (const std::vector< ArrayCpuGpu< T > * > &bottom, const std::vector< ArrayCpuGpu< T > * > &top, const int maxPeaks, const int outputChannels=-1, const int gpuID=0) |
|
virtual const char * | type () const |
|
void | setThreshold (const T threshold) |
|
void | setOffset (const Point< T > &offset) |
|
virtual void | Forward (const std::vector< ArrayCpuGpu< T > * > &bottom, const std::vector< ArrayCpuGpu< T > * > &top) |
|
virtual void | Forward_cpu (const std::vector< ArrayCpuGpu< T > * > &bottom, const std::vector< ArrayCpuGpu< T > * > &top) |
|
virtual void | Forward_gpu (const std::vector< ArrayCpuGpu< T > * > &bottom, const std::vector< ArrayCpuGpu< T > * > &top) |
|
virtual void | Forward_ocl (const std::vector< ArrayCpuGpu< T > * > &bottom, const std::vector< ArrayCpuGpu< T > * > &top) |
|
virtual void | Backward_cpu (const std::vector< ArrayCpuGpu< T > * > &top, const std::vector< bool > &propagate_down, const std::vector< ArrayCpuGpu< T > * > &bottom) |
|
virtual void | Backward_gpu (const std::vector< ArrayCpuGpu< T > * > &top, const std::vector< bool > &propagate_down, const std::vector< ArrayCpuGpu< T > * > &bottom) |
|
template<typename T>
class op::NmsCaffe< T >
Definition at line 12 of file nmsCaffe.hpp.