Downloads · 30 days
0
doggywastaken/leaky_vnet
leaky_vnet is a image segmentation model from doggywastaken. Use it for the image segmentation task on the model card, and read the license before you ship it in a product. The card lists the license as mit.
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.
Downloads · 30 days
0
Access
Public
Updated May 20, 2024
Repo size
—
Likes
0
Public
Click a slice to open those files.
.py6.9 KB · 68%
From the Hugging Face model README
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)