python code for nms_gpu_code

model/utils/nms/_nms_gpu_post_py.py

+
+import numpy as np
+
+def _nms_gpu_post( mask,
+                  n_bbox,
+                   threads_per_block,
+                   col_blocks
+                  ):
+
+    selection = np.zeros((n_bbox,), dtype=np.int32)
+    remv = np.zeros((col_blocks,), dtype=np.uint64)
+
+    for i in range(n_bbox):
+        nblock = i // threads_per_block
+        inblock = i % threads_per_block
+
+        if not (remv[nblock] & one_ull << inblock):
+            selection[n_selection] = i
+            n_selection += 1
+
+            index = i * col_blocks
+            for j in range(nblock, col_blocks):
+                remv[j] |= mask[index + j]
+    return selection, n_selection