From f35faf5f9dba7ef5d81cba31e7001102f2a0b37a Mon Sep 17 00:00:00 2001
From: goolhanrry <goolhanrry@gmail.com>
Date: Mon, 16 Dec 2019 17:49:18 +0800
Subject: [PATCH] Fix loader for images with no proposals
---
mmdet/datasets/pipelines/loading.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/mmdet/datasets/pipelines/loading.py b/mmdet/datasets/pipelines/loading.py
index f3aee9f..6a9938c 100644
--- a/mmdet/datasets/pipelines/loading.py
+++ b/mmdet/datasets/pipelines/loading.py
@@ -145,7 +145,7 @@ class LoadProposals(object):
proposals = proposals[:self.num_max_proposals]
if len(proposals) == 0:
- proposals = np.array([0, 0, 0, 0], dtype=np.float32)
+ proposals = np.array([[0, 0, 0, 0]], dtype=np.float32)
results['proposals'] = proposals
results['bbox_fields'].append('proposals')
return results
--
GitLab