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