A small bug in concat_batcher.py #673

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'))