OpenPCDet icon indicating copy to clipboard operation
OpenPCDet copied to clipboard

Questions about functions mask_points_and_boxes_outside_range

Open LandDreamer opened this issue 1 year ago • 0 comments

In data_processor.py

    def mask_points_and_boxes_outside_range(self, data_dict=None, config=None):
        if data_dict is None:
            return partial(self.mask_points_and_boxes_outside_range, config=config)

        if data_dict.get('points', None) is not None:
            mask = common_utils.mask_points_by_range(data_dict['points'], self.point_cloud_range)
            data_dict['points'] = data_dict['points'][mask]

        if data_dict.get('gt_boxes', None) is not None and config.REMOVE_OUTSIDE_BOXES and self.training:
            mask = box_utils.mask_boxes_outside_range_numpy(
                data_dict['gt_boxes'], self.point_cloud_range, min_num_corners=config.get('min_num_corners', 1), 
                use_center_to_filter=config.get('USE_CENTER_TO_FILTER', True)
            )
            data_dict['gt_boxes'] = data_dict['gt_boxes'][mask]
        return data_dict

It seems that missing mask gt_names, as gt_names and gt_boxes should be aligned

 data_dict['gt_boxes'] = data_dict['gt_boxes'][mask]
 data_dict['gt_names'] = data_dict['gt_names'][mask]

LandDreamer avatar May 07 '24 18:05 LandDreamer