DynamicScatter¶
- class mmcv.ops.DynamicScatter(voxel_size: List, point_cloud_range: List, average_points: bool)[源代码]¶
Scatters points into voxels, used in the voxel encoder with dynamic voxelization.
注解
The CPU and GPU implementation get the same output, but have numerical difference after summation and division (e.g., 5e-7).
- 参数
- forward(points: torch.Tensor, coors: torch.Tensor) → Tuple[torch.Tensor, torch.Tensor][源代码]¶
Scatters points/features into voxels.
- 参数
points (torch.Tensor) – Points to be reduced into voxels.
coors (torch.Tensor) – Corresponding voxel coordinates (specifically multi-dim voxel index) of each points.
- 返回
A tuple contains two elements. The first one is the voxel features with shape [M, C] which are respectively reduced from input features that share the same voxel coordinates. The second is voxel coordinates with shape [M, ndim].
- 返回类型
- forward_single(points: torch.Tensor, coors: torch.Tensor) → Tuple[torch.Tensor, torch.Tensor][源代码]¶
Scatters points into voxels.
- 参数
points (torch.Tensor) – Points to be reduced into voxels.
coors (torch.Tensor) – Corresponding voxel coordinates (specifically multi-dim voxel index) of each points.
- 返回
A tuple contains two elements. The first one is the voxel features with shape [M, C] which are respectively reduced from input features that share the same voxel coordinates. The second is voxel coordinates with shape [M, ndim].
- 返回类型