gpu_nms.hpp 180 Bytes
Newer Older
Kai Chen's avatar
Kai Chen committed
1
2
3
void _nms(int* keep_out, int* num_out, const float* boxes_host, int boxes_num,
          int boxes_dim, float nms_overlap_thresh, int device_id, size_t base);
size_t nms_Malloc();