diff --git a/mmdet/datasets/pipelines/loading.py b/mmdet/datasets/pipelines/loading.py index f3aee9fb3029653978b4a1f4de6810638f76bcc5..6a9938c71e6688594b9867d46efd04dc45d628f7 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