Shortcuts

RoIPointPool3d

class mmcv.ops.RoIPointPool3d(num_sampled_points: int = 512)[源代码]

Encode the geometry-specific features of each 3D proposal.

Please refer to Paper of PartA2 for more details.

参数

num_sampled_points (int, optional) – Number of samples in each roi. Default: 512.

forward(points: torch.Tensor, point_features: torch.Tensor, boxes3d: torch.Tensor)Tuple[torch.Tensor][源代码]
参数
  • points (torch.Tensor) – Input points whose shape is (B, N, C).

  • point_features (torch.Tensor) – Features of input points whose shape is (B, N, C).

  • boxes3d (B, M, 7), Input bounding boxes whose shape is (B, M, 7) –

返回

A tuple contains two elements. The first one is the pooled features whose shape is (B, M, 512, 3 + C). The second is an empty flag whose shape is (B, M).

返回类型

tuple[torch.Tensor]

Read the Docs v: latest
Versions
latest
stable
2.x
v2.1.0
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.