From 1f4177bd6d07dbe2fe602e831f1bb8c7b37004f1 Mon Sep 17 00:00:00 2001 From: Wenwei Zhang <40779233+ZwwWayne@users.noreply.github.com> Date: Wed, 19 Feb 2020 10:37:50 +0800 Subject: [PATCH] fix (get_cls_results): use np.empty for empty bbox rather than np.arrary (#2116) --- mmdet/core/evaluation/mean_ap.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mmdet/core/evaluation/mean_ap.py b/mmdet/core/evaluation/mean_ap.py index 4e3cd5d0765..fedc51d6313 100644 --- a/mmdet/core/evaluation/mean_ap.py +++ b/mmdet/core/evaluation/mean_ap.py @@ -258,7 +258,7 @@ def get_cls_results(det_results, annotations, class_id): ignore_inds = ann['labels_ignore'] == (class_id + 1) cls_gts_ignore.append(ann['bboxes_ignore'][ignore_inds, :]) else: - cls_gts_ignore.append(np.array((0, 4), dtype=np.float32)) + cls_gts_ignore.append(np.empty((0, 4), dtype=np.float32)) return cls_dets, cls_gts, cls_gts_ignore