diff options
| author | carp <25677564+carp@users.noreply.github.com> | 2020-07-13 13:27:55 -0400 |
|---|---|---|
| committer | carp <25677564+carp@users.noreply.github.com> | 2020-07-13 13:27:55 -0400 |
| commit | cb961ed0d6896309336159bda34ed2f75fb60a84 (patch) | |
| tree | 6d75fcd9bd32b2e884e299ea97e361513036b3a2 /anime-face-detector/nms/gpu_nms.hpp | |
| parent | e78f351e06e432a607ebadc8de3ea6d27315c088 (diff) | |
| download | yaoi-communism-cb961ed0d6896309336159bda34ed2f75fb60a84.tar.gz yaoi-communism-cb961ed0d6896309336159bda34ed2f75fb60a84.zip | |
add face detection
Diffstat (limited to 'anime-face-detector/nms/gpu_nms.hpp')
| -rw-r--r-- | anime-face-detector/nms/gpu_nms.hpp | 2 |
1 files changed, 2 insertions, 0 deletions
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); |