Spaces:
Running
Running
gavinyuan
commited on
Commit
·
277bad0
1
Parent(s):
69059e0
update: PIPNet
Browse files
third_party/PIPNet/FaceBoxesV2/utils/nms/cpu_nms.pyx
CHANGED
@@ -25,8 +25,7 @@ def cpu_nms(np.ndarray[np.float32_t, ndim=2] dets, np.float thresh):
|
|
25 |
cdef np.ndarray[np.int_t, ndim=1] order = scores.argsort()[::-1]
|
26 |
|
27 |
cdef int ndets = dets.shape[0]
|
28 |
-
cdef np.ndarray[np.int_t, ndim=1] suppressed =
|
29 |
-
np.zeros((ndets), dtype=np.int32)
|
30 |
|
31 |
# nominal indices
|
32 |
cdef int _i, _j
|
|
|
25 |
cdef np.ndarray[np.int_t, ndim=1] order = scores.argsort()[::-1]
|
26 |
|
27 |
cdef int ndets = dets.shape[0]
|
28 |
+
cdef np.ndarray[np.int_t, ndim=1] suppressed = np.zeros((ndets), dtype=int)
|
|
|
29 |
|
30 |
# nominal indices
|
31 |
cdef int _i, _j
|