layers
Reusable neural network blocks shared across the model architectures.
Modules:
-
act–Activation factory wrapper around
torch_geometric.nn.resolver. -
affine–Per-channel affine transformation as a function and a learnable module.
-
anchors–Anchor-based dense detection heads for the voxel detectors (PointPillars, SECOND).
-
bev_backbone–SSD-style 2D BEV backbones shared by the voxel detectors (PointPillars, SECOND, Voxel Mamba).
-
conv2d_blocks–2D convolution block with optional normalization and activation.
-
conv3d_blocks–3D convolution block with optional normalization and activation.
-
dropouts–Dropout layers: stochastic depth (
DropPath) and thecreate_dropoutfactory. -
fps–Farthest point sampling (FPS) as a module returning sampled indices.
-
geometric_affine–PointMLP grouping convolution with geometric affine neighborhood normalization.
-
grid_pool–Grid-based point cloud pooling (downsampling).
-
linear_blocks–Linear block with optional normalization and activation.
-
norms–Normalization factory wrapper with spatial dimensionality support.
-
octree_attention–OctFormer octree window attention with relative position encoding.
-
octree_blocks–Octree convolution and deconvolution blocks with normalization and activation.
-
pdnorm–Prompt-driven normalization (PDNorm) routing each batch through a per-condition norm.
-
point_patch_embed–Mini-PointNet patch embedding turning local point groups into tokens.
-
pointconv–PointConv message-passing convolutions, with and without density reweighting.
-
pointconv_sa–PointConv set abstraction modules for hierarchical and global feature extraction.
-
pointnet2_blocks–PointNet++ set abstraction and feature propagation blocks.
-
pointnext_blocks–PointNeXt convolution layer introduced in the
-
pools–Per-segment pooling modules over packed batches and the
create_pool/create_adaptive_poolfactories. -
pvcnn_blocks–PVCNN building blocks: voxelization, squeeze-and-excitation, and point-voxel convolution.
-
rope–3D rotary position embedding for point-cloud attention.
-
serialized_attention–Serialized attention variants used by Point Transformer V3 and descendants.
-
serialized_pool–Pooling and upsampling driven by point cloud serialization codes.
-
spconv_blocks–Sparse convolution blocks: submanifold, strided, and residual variants.
-
tnet–T-Net alignment modules predicting affine transforms for point and feature spaces.
-
transformer–Standard pre-norm transformer building blocks for point-cloud backbones.
-
vfe–Dynamic voxel feature encoder shared by the voxel detectors (Voxel Mamba, LION).
-
view–Tensor reshaping as a module wrapping
Tensor.view. -
xconv–PointCNN XConv layer: convolution on X-transformed point neighborhoods.