Leaky V-Net in Pytorch

This is a fork of @Dootmaan's VNet.PyTorch repo, who attempted a faithful recreation of the original V-Net: Fully Convolutional Neural Network for Volumetric Medical Image paper, with as little adaptations as possible.

This repo's model has some minor adaptations to fit it's designated application:

  • ReLU layers is now the leaky version to allow for more consistent convergence on small training datasets

  • Final Sigmoid layer has been removed from the network in favour of manual thresholding (for flexibility during testing)

Downloads last month

-

Downloads are not tracked for this model. How to track
Inference Providers NEW
This model is not currently available via any of the supported Inference Providers.
The model cannot be deployed to the HF Inference API: The model has no library tag.