3d Convolutional Network Python Packages

minkowskiengine

Minkowski Engine is an auto-diff neural network library for high-dimensional sparse tensors

timm-3d

PyTorch Volume Models for 3D data

uniformer-pytorch

Implementation of Uniformer, a simple attention and 3d convolutional net that achieved SOTA in a number of video classification tasks, debuted in ICLR 2022

elektronn3

A PyTorch-based library for working with 3D and 2D convolutional neural networks, with focus on semantic segmentation of volumetric biomedical image data