cpu nms

yushenxiang · Nov 26, 2017 · 93bb345 · 93bb345
1 parent 7cd8e88
commit 93bb345
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/test_RFB.py b/test_RFB.py
@@ -110,7 +110,7 @@ def test_net(save_folder, net, detector, cuda, testset, transform, max_per_image
             c_dets = np.hstack((c_bboxes, c_scores[:, np.newaxis])).astype(
                 np.float32, copy=False)
 
-            keep = nms(c_dets, 0.45)
+            keep = nms(c_dets, 0.45, force_cpu=True)
             keep = keep[:50]
             c_dets = c_dets[keep, :]
             all_boxes[j][i] = c_dets