Open3D-ML
Open3D-ML copied to clipboard
Is there a way to filter ground truth 3D bounding boxes by distance to ego, yaw, confidence etc
Checklist
- [X] I have searched for similar issues.
- [X] I have tested with the latest development wheel.
- [X] I have checked the release documentation and the latest documentation (for
masterbranch).
Proposed new feature or change
Right now for 3D bounding boxes we can view and filter them in tensorboard by classes. Ie:
writer.add_3d("object_detection", {"bboxes": data['bounding_boxes']},
step,
label_to_names=dset.get_label_to_names())
It would be super useful if we could also have a frontend tensorboard toggle feature to filter boxes by distance, yaw, confidence of prediction etc.
writer.add_3d("object_detection", {"bboxes": data['bounding_boxes']},
step,
label_to_names=dset.get_label_to_names(),
distance=distance_to_boxes,
yaw=yaw_to_boxes,
confidence=confidence_of_boxes
)
The filter by classes feature is already super useful for debugging and visualizing which classes my neural network is underperforming. Would be amazing if we could have a visualizer across these other fields as well!