"tests/vscode:/vscode.git/clone" did not exist on "0ce92a86ed3e721cdc2874f17c2300f6753886ba"
Commit f35faf5f authored by goolhanrry's avatar goolhanrry
Browse files

Fix loader for images with no proposals

parent e2c4ea3b
......@@ -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
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment