metadata

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)