Spaces:

ygtxr1997
/

ReliableSwap_Demo

Running

gavinyuan commited on Jun 6, 2023

Commit

69059e0

1 Parent(s): c69e759

update: PIPNet

Files changed (1) hide show

third_party/PIPNet/FaceBoxesV2/utils/nms/cpu_nms.pyx CHANGED Viewed

@@ -26,7 +26,7 @@ def cpu_nms(np.ndarray[np.float32_t, ndim=2] dets, np.float thresh):
     cdef int ndets = dets.shape[0]
     cdef np.ndarray[np.int_t, ndim=1] suppressed = \
-            np.zeros((ndets), dtype=np.int)
     # nominal indices
     cdef int _i, _j

     cdef int ndets = dets.shape[0]
     cdef np.ndarray[np.int_t, ndim=1] suppressed = \
+            np.zeros((ndets), dtype=np.int32)
     # nominal indices
     cdef int _i, _j