Shortcuts

PointsSampler

class mmcv.ops.PointsSampler(num_point: List[int], fps_mod_list: List[str] = ['D-FPS'], fps_sample_range_list: List[int] = [- 1])[源代码]

Points sampling.

参数
  • num_point (list[int]) – Number of sample points.

  • fps_mod_list (list[str], optional) – Type of FPS method, valid mod [‘F-FPS’, ‘D-FPS’, ‘FS’], Default: [‘D-FPS’]. F-FPS: using feature distances for FPS. D-FPS: using Euclidean distances of points for FPS. FS: using F-FPS and D-FPS simultaneously.

  • fps_sample_range_list (list[int], optional) – Range of points to apply FPS. Default: [-1].

forward(points_xyz: torch.Tensor, features: torch.Tensor)torch.Tensor[源代码]
参数
  • points_xyz (torch.Tensor) – (B, N, 3) xyz coordinates of the points.

  • features (torch.Tensor) – (B, C, N) features of the points.

返回

(B, npoint, sample_num) Indices of sampled points.

返回类型

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.