Unverified Commit ea32fe67 authored by Alex Pasquali's avatar Alex Pasquali Committed by GitHub
Browse files

[Fix] Fixed bug on Custom3DDataset (#1519)

* Fixed bug on Custom3DDataset

The func 'get_ann_info' was supposed to return a dict with the following keys:
- gt_bboxes_3d
- gt_labels_3d
- gt_names
but gt_names wasn't included in the return dict.

I added this key-value couple to the return dict of 'get_ann_info'

* fixed code linting
parent d5854460
......@@ -182,7 +182,9 @@ class Custom3DDataset(Dataset):
origin=(0.5, 0.5, 0.5)).convert_to(self.box_mode_3d)
anns_results = dict(
gt_bboxes_3d=gt_bboxes_3d, gt_labels_3d=gt_labels_3d)
gt_bboxes_3d=gt_bboxes_3d,
gt_labels_3d=gt_labels_3d,
gt_names=gt_names_3d)
return anns_results
def pre_pipeline(self, 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