---
license: mit
datasets:
- doggywastaken/manual_breast_segs
language:
- en
pipeline_tag: image-segmentation
---
# 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)