Shortcuts

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

参数
  • voxel_size (list) – list [x, y, z] size of three dimension.

  • point_cloud_range (list) – The coordinate range of points, [x_min, y_min, z_min, x_max, y_max, z_max].

  • average_points (bool) – whether to use avg pooling to scatter points into voxel.

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].

返回类型

tuple[torch.Tensor]

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].

返回类型

tuple[torch.Tensor]

Read the Docs v: stable
Versions
latest
stable
2.x
v2.0.1
v2.0.0
1.x
v1.7.1
v1.7.0
v1.6.2
v1.6.1
v1.6.0
v1.5.3
v1.5.2_a
v1.5.1
v1.5.0
v1.4.8
v1.4.7
v1.4.6
v1.4.5
v1.4.4
v1.4.3
v1.4.2
v1.4.1
v1.4.0
v1.3.18
v1.3.17
v1.3.16
v1.3.15
v1.3.14
v1.3.13
Downloads
pdf
html
epub
On Read the Docs
Project Home
Builds

Free document hosting provided by Read the Docs.