|
| | 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.