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 ...

This is a 3D model I adapted and optimized for 3D from the 2D Unet at https://github.com/milesial/Pytorch-UNet. I've used this for 3D segmentation and also pose ...