-
Notifications
You must be signed in to change notification settings - Fork 351
Open
Description
In the concat_batcher.py file, in the implementation of the class ObjectDetectBatch,
class ObjectDetectBatch:
def __init__(self, batches):
"""Initialize.
Args:
batches: A batch of data
Returns:
class: The corresponding class.
"""
self.point = []
self.labels = []
self.bboxes = []
self.bbox_objs = []
self.calib = []
self.attr = []
for batch in batches:
data = batch['data']
self.point.append(torch.tensor(data['point'], dtype=torch.float32))
self.labels.append(
torch.tensor(data['labels'], dtype=torch.int64) if 'labels' in
data else None)
if len(data.get('bboxes', [])) > 0:
self.bboxes.append(
torch.tensor(data['bboxes'], dtype=torch.float32
) if 'bboxes' in data else None)
else:
self.bboxes.append(torch.zeros((0, 7)))
self.bbox_objs.append(data.get('bbox_objs'))
self.calib.append(data.get('calib'))
shouldn't we add a line
self.attr.append(batch['attr'])
in the end of the __init__ function? Otherwise the 'attr' field is missing, and it causes save_test_result() function in kitti.py not saving anything.
Metadata
Metadata
Assignees
Labels
No labels