OpenPose 1.0.0rc2 OpenPose: A Real-Time Multi-Person Key-Point Detection And Multi-Threading C++ Library

- Main Page
- Namespaces
- Classes
- Files

- File List
- File Members

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
4 #include <openpose/core/common.hpp>
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, uint8_t* kernelGpuPtr, uint8_t* kernelCpuPtr, 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
op::nmsCpu
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)
op::nmsOcl
void nmsOcl(T *targetPtr, uint8_t *kernelGpuPtr, uint8_t *kernelCpuPtr, 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)
common.hpp
op::nmsGpu
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)