Skip to content
Snippets Groups Projects
Unverified Commit 0f59c397 authored by Kai Chen's avatar Kai Chen Committed by GitHub
Browse files

Merge pull request #1819 from goolhanrry/master

Fix loader for images with no proposals
parents e2c4ea3b f35faf5f
No related branches found
No related tags found
No related merge requests found
...@@ -145,7 +145,7 @@ class LoadProposals(object): ...@@ -145,7 +145,7 @@ class LoadProposals(object):
proposals = proposals[:self.num_max_proposals] proposals = proposals[:self.num_max_proposals]
if len(proposals) == 0: 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['proposals'] = proposals
results['bbox_fields'].append('proposals') results['bbox_fields'].append('proposals')
return results return results
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment