3D Sparse Voxel Autoencoder

Inspired by the machine learning research I did, I created a 3D autoencoder model using the spconv framework with PyTorch. The key innovation is training a sparse model, because most of a voxelized model is empty. This massively decreases VRAM usage and increases training speed.

Python PyTorch Data ML Training
CV Test

3D Sparse Autoencoder

Figure 1: Model Architecture.

Description