File size: 1,014 Bytes
7fc5a59 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 | #ifndef OPENPOSE_NET_NMS_BASE_HPP
#define OPENPOSE_NET_NMS_BASE_HPP
#include <openpose/core/common.hpp>
namespace op
{
template <typename T>
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);
// Windows: Cuda functions do not include OP_API
template <typename T>
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);
// Windows: OpenCL functions do not include OP_API
template <typename T>
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);
}
#endif // OPENPOSE_NET_NMS_BASE_HPP
|