File size: 671 Bytes
70f7a9d d26715f 70f7a9d |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 |
---
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) |