diff options
| author | carp <25677564+carp@users.noreply.github.com> | 2020-07-13 13:40:11 -0400 |
|---|---|---|
| committer | carp <25677564+carp@users.noreply.github.com> | 2020-07-13 13:40:11 -0400 |
| commit | 50e411320563894d411b0c37d37cb16105a908af (patch) | |
| tree | 01e1a43e11f9e51cf6a46d9a3d40814b76ad6d7b /anime-face-detector/nms/gpu_nms.pyx | |
| parent | f457064bb15a00010959e664492d87f3bfe82537 (diff) | |
| download | yaoi-communism-50e411320563894d411b0c37d37cb16105a908af.tar.gz yaoi-communism-50e411320563894d411b0c37d37cb16105a908af.zip | |
removing submodule
Diffstat (limited to 'anime-face-detector/nms/gpu_nms.pyx')
| -rw-r--r-- | anime-face-detector/nms/gpu_nms.pyx | 31 |
1 files changed, 0 insertions, 31 deletions
diff --git a/anime-face-detector/nms/gpu_nms.pyx b/anime-face-detector/nms/gpu_nms.pyx deleted file mode 100644 index 55878db..0000000 --- a/anime-face-detector/nms/gpu_nms.pyx +++ /dev/null @@ -1,31 +0,0 @@ -# -------------------------------------------------------- -# Faster R-CNN -# Copyright (c) 2015 Microsoft -# Licensed under The MIT License [see LICENSE for details] -# Written by Ross Girshick -# -------------------------------------------------------- - -import numpy as np -cimport numpy as np - -assert sizeof(int) == sizeof(np.int32_t) - -cdef extern from "gpu_nms.hpp": - void _nms(np.int32_t*, int*, np.float32_t*, int, int, float, int) - -def gpu_nms(np.ndarray[np.float32_t, ndim=2] dets, np.float thresh, - np.int32_t device_id=0): - cdef int boxes_num = dets.shape[0] - cdef int boxes_dim = dets.shape[1] - cdef int num_out - cdef np.ndarray[np.int32_t, ndim=1] \ - keep = np.zeros(boxes_num, dtype=np.int32) - cdef np.ndarray[np.float32_t, ndim=1] \ - scores = dets[:, 4] - cdef np.ndarray[np.int64_t, ndim=1] \ - order = scores.argsort()[::-1] - cdef np.ndarray[np.float32_t, ndim=2] \ - sorted_dets = dets[order, :] - _nms(&keep[0], &num_out, &sorted_dets[0, 0], boxes_num, boxes_dim, thresh, device_id) - keep = keep[:num_out] - return list(order[keep]) |