fix bug in _write_voc_results_file. When dets is empty Megvii-BaseDet…

…ection#1755
vossr · Apr 15, 2024 · 68e0286 · 68e0286
1 parent 2360277
commit 68e0286
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/yolox/data/datasets/voc.py b/yolox/data/datasets/voc.py
@@ -261,7 +261,7 @@ def _write_voc_results_file(self, all_boxes):
                 for im_ind, index in enumerate(self.ids):
                     index = index[1]
                     dets = all_boxes[cls_ind][im_ind]
-                    if dets == []:
+                    if dets.shape[0] == 0:
                         continue
                     for k in range(dets.shape[0]):
                         f.write(