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  OP_API void nmsCpu(T* targetPtr, int* kernelPtr, const T* const sourcePtr, const T threshold,
10  const std::array<int, 4>& targetSize, const std::array<int, 4>& sourceSize,
11  const Point<T>& offset);
12 
13  template <typename T>
14  OP_API void nmsGpu(T* targetPtr, int* kernelPtr, const T* const sourcePtr, const T threshold,
15  const std::array<int, 4>& targetSize, const std::array<int, 4>& sourceSize,
16  const Point<T>& offset);
17 
18  template <typename T>
19  OP_API void nmsOcl(T* targetPtr, int* kernelPtr, const T* const sourcePtr, const T threshold,
20  const std::array<int, 4>& targetSize, const std::array<int, 4>& sourceSize,
21  const Point<T>& offset, const int gpuID = 0);
22 }
23 
24 #endif // OPENPOSE_NET_NMS_BASE_HPP
OP_API 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)
OP_API 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)
OP_API 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)
#define OP_API
Definition: macros.hpp:18