Python library with Neural Networks for Volume (3D) Segmentation based on PyTorch. This library is based on famous Segmentation Models Pytorch library for images. Most of the documentation can be used ...

I've used this for 3D segmentation and also pose detection (with MSEloss) tasks with surprising success. To give an idea of what you can achieve, I can use a batch size of around 10 on a 32 GB GPU ...