nn.functional.point_pillars_preprocess

horizon_plugin_pytorch.nn.functional.point_pillars_preprocess(points_list: List[Tensor], pc_range: Tensor, voxel_size: Tensor, max_voxels: int, max_points_per_voxel: int, use_max: bool, norm_range: Tensor, norm_dims: Tensor)

Preprocess PointPillars.

  • Parameters:
    • points_list – [(M1, ndim), (M2, ndim),…], List of PointCloud data.
    • pc_range – (6,), indicate voxel range, format: [x_min, y_min, z_min, x_max, y_max, z_max]
    • voxel_size – (3,), xyz, indicate voxel size.
    • max_voxels – Indicate maximum voxels.
    • max_points_per_voxel – Indicate maximum points contained in a voxel.
    • use_max – Whether to use max_voxels, for deploy should be True.
    • norm_range – Feature range, like [x_min, y_min, z_min, …, x_max, y_max, z_max, …].
    • norm_dims – Dims to do normalize.
  • Returns: (features, coords), encoded feature and coordinates in (idx, z, y, x) format.
  • Return type: (Tensor, Tensor)