- class mmcv.ops.RoIAwarePool3d(out_size: Union[int, tuple], max_pts_per_voxel: int = 128, mode: str = 'max')¶
Encode the geometry-specific features of each 3D proposal.
Please refer to PartA2 for more details.
- forward(rois: torch.Tensor, pts: torch.Tensor, pts_feature: torch.Tensor) → torch.Tensor¶
Pooled features whose shape is [N, out_x, out_y, out_z, C].
- Return type