From cb961ed0d6896309336159bda34ed2f75fb60a84 Mon Sep 17 00:00:00 2001 From: carp <25677564+carp@users.noreply.github.com> Date: Mon, 13 Jul 2020 13:27:55 -0400 Subject: add face detection --- anime-face-detector/nms/gpu_nms.hpp | 2 ++ 1 file changed, 2 insertions(+) create mode 100644 anime-face-detector/nms/gpu_nms.hpp (limited to 'anime-face-detector/nms/gpu_nms.hpp') diff --git a/anime-face-detector/nms/gpu_nms.hpp b/anime-face-detector/nms/gpu_nms.hpp new file mode 100644 index 0000000..68b6d42 --- /dev/null +++ b/anime-face-detector/nms/gpu_nms.hpp @@ -0,0 +1,2 @@ +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); -- cgit v1.2.3