diff --git a/yolox/data/datasets/voc.py b/yolox/data/datasets/voc.py index bdacd8019..cae2c3eba 100644 --- 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(