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
nmsBase.hpp
Go to the documentation of this file.
1 #ifndef OPENPOSE_NET_NMS_BASE_HPP
2 #define OPENPOSE_NET_NMS_BASE_HPP
3 
5 
6 namespace op
7 {
8  template <typename T>
9  void nmsCpu(
10  T* targetPtr, int* kernelPtr, const T* const sourcePtr, const T threshold, const std::array<int, 4>& targetSize,
11  const std::array<int, 4>& sourceSize, const Point<T>& offset);
12 
13  // Windows: Cuda functions do not include OP_API
14  template <typename T>
15  void nmsGpu(
16  T* targetPtr, int* kernelPtr, const T* const sourcePtr, const T threshold, const std::array<int, 4>& targetSize,
17  const std::array<int, 4>& sourceSize, const Point<T>& offset);
18 
19  // Windows: OpenCL functions do not include OP_API
20  template <typename T>
21  void nmsOcl(
22  T* targetPtr, int* kernelPtr, const T* const sourcePtr, const T threshold, const std::array<int, 4>& targetSize,
23  const std::array<int, 4>& sourceSize, const Point<T>& offset, const int gpuID = 0);
24 }
25 
26 #endif // OPENPOSE_NET_NMS_BASE_HPP
void nmsOcl(T *targetPtr, int *kernelPtr, const T *const sourcePtr, const T threshold, const std::array< int, 4 > &targetSize, const std::array< int, 4 > &sourceSize, const Point< T > &offset, const int gpuID=0)
void nmsCpu(T *targetPtr, int *kernelPtr, const T *const sourcePtr, const T threshold, const std::array< int, 4 > &targetSize, const std::array< int, 4 > &sourceSize, const Point< T > &offset)
void nmsGpu(T *targetPtr, int *kernelPtr, const T *const sourcePtr, const T threshold, const std::array< int, 4 > &targetSize, const std::array< int, 4 > &sourceSize, const Point< T > &offset)