open3d.ml.torch.layers#

High level layer API for building networks.

This module contains layers for processing 3D data. All layers subclass torch.nn.Module

Classes

ContinuousConv(in_channels, filters, kernel_size)

Continuous Convolution.

FixedRadiusSearch([metric, ...])

Fixed radius search for 3D point clouds.

KNNSearch([metric, ignore_query_point, ...])

KNN search for 3D point clouds.

RadiusSearch([metric, ignore_query_point, ...])

Radius search for 3D point clouds.

SparseConv(in_channels, filters, kernel_size)

Sparse Convolution.

SparseConvTranspose(in_channels, filters, ...)

Sparse Transposed Convolution.

VoxelPooling([position_fn, feature_fn])

Voxel pooling for 3D point clouds.