GitPlanet

Cheap and reliable Node.js hosting starts at $3/month, and $1/month static HTML hosting

Created with love in Canada, visit hostnodejs.com today

Feel like to post an Ad? Learn Details
All Projects → mattmacy → Vnet.pytorch

mattmacy / Vnet.pytorch

Licence: bsd-3-clause
A PyTorch implementation for V-Net: Fully Convolutional Neural Networks for Volumetric Medical Image Segmentation

Programming Languages

python
139335 projects - #7 most used programming language

Labels

pytorchconvolutional-neural-networkssemantic-segmentationfully-convolutional-networks

Projects that are alternatives of or similar to Vnet.pytorch

3dunet abdomen cascade
Stars: ✭ 91 (-82.02%)
Mutual labels:  convolutional-neural-networks, semantic-segmentation, fully-convolutional-networks
semantic segmentation
Semantically segment the road in the given image.
Stars: ✭ 91 (-82.02%)
Mutual labels:  semantic-segmentation, fully-convolutional-networks
atomai
Deep and Machine Learning for Microscopy
Stars: ✭ 77 (-84.78%)
Mutual labels:  semantic-segmentation, fully-convolutional-networks
Espnet
ESPNet: Efficient Spatial Pyramid of Dilated Convolutions for Semantic Segmentation
Stars: ✭ 473 (-6.52%)
Mutual labels:  convolutional-neural-networks, semantic-segmentation
Ynet
Y-Net: Joint Segmentation and Classification for Diagnosis of Breast Biopsy Images
Stars: ✭ 100 (-80.24%)
Mutual labels:  convolutional-neural-networks, semantic-segmentation
Self Driving Golf Cart
Be Driven 🚘
Stars: ✭ 147 (-70.95%)
Mutual labels:  convolutional-neural-networks, semantic-segmentation
Bonnet
Bonnet: An Open-Source Training and Deployment Framework for Semantic Segmentation in Robotics.
Stars: ✭ 274 (-45.85%)
Mutual labels:  convolutional-neural-networks, semantic-segmentation
Chainer Pspnet
PSPNet in Chainer
Stars: ✭ 76 (-84.98%)
Mutual labels:  convolutional-neural-networks, semantic-segmentation
Geospatial Machine Learning
A curated list of resources focused on Machine Learning in Geospatial Data Science.
Stars: ✭ 289 (-42.89%)
Mutual labels:  convolutional-neural-networks, semantic-segmentation
Pytorch Semseg
Semantic Segmentation Architectures Implemented in PyTorch
Stars: ✭ 3,180 (+528.46%)
Mutual labels:  semantic-segmentation, fully-convolutional-networks
Cascaded Fcn
Source code for the MICCAI 2016 Paper "Automatic Liver and Lesion Segmentation in CT Using Cascaded Fully Convolutional NeuralNetworks and 3D Conditional Random Fields"
Stars: ✭ 296 (-41.5%)
Mutual labels:  semantic-segmentation, fully-convolutional-networks
Cardiac Segmentation
Convolutional Neural Networks for Cardiac Segmentation
Stars: ✭ 94 (-81.42%)
Mutual labels:  convolutional-neural-networks, fully-convolutional-networks
Espnetv2
A light-weight, power efficient, and general purpose convolutional neural network
Stars: ✭ 377 (-25.49%)
Mutual labels:  convolutional-neural-networks, semantic-segmentation
Fashion-Clothing-Parsing
FCN, U-Net models implementation in TensorFlow for fashion clothing parsing
Stars: ✭ 29 (-94.27%)
Mutual labels:  semantic-segmentation, fully-convolutional-networks
Elektronn3
A PyTorch-based library for working with 3D and 2D convolutional neural networks, with focus on semantic segmentation of volumetric biomedical image data
Stars: ✭ 78 (-84.58%)
Mutual labels:  convolutional-neural-networks, semantic-segmentation
FCNN-example
This is a fully convolutional neural net exercise to detect houses from aerial images.
Stars: ✭ 28 (-94.47%)
Mutual labels:  semantic-segmentation, fully-convolutional-networks
Minkowskiengine
Minkowski Engine is an auto-diff neural network library for high-dimensional sparse tensors
Stars: ✭ 1,110 (+119.37%)
Mutual labels:  convolutional-neural-networks, semantic-segmentation
Espnetv2 Coreml
Semantic segmentation on iPhone using ESPNetv2
Stars: ✭ 66 (-86.96%)
Mutual labels:  convolutional-neural-networks, semantic-segmentation
Fcn Pytorch
🚘 Easiest Fully Convolutional Networks
Stars: ✭ 278 (-45.06%)
Mutual labels:  semantic-segmentation, fully-convolutional-networks
Bcdu Net
BCDU-Net : Medical Image Segmentation
Stars: ✭ 314 (-37.94%)
Mutual labels:  convolutional-neural-networks, semantic-segmentation
View All Similar Projects ➔

A PyTorch implementation of V-Net

Vnet is a PyTorch implementation of the paper V-Net: Fully Convolutional Neural Networks for Volumetric Medical Image Segmentation by Fausto Milletari, Nassir Navab, and Seyed-Ahmad Ahmadi. Although this implementation is still a work in progress, I'm seeing a respectable 0.355% test error rate and a Dice coefficient of .9825 segmenting lungs from the LUNA16 data set after 249 epochs. The official implementation is available in the faustomilletari/VNet repo on GitHub.

This implementation relies on the LUNA16 loader and dice loss function from the Torchbiomed package.

Differences with the official version

This version uses batch normalization and dropout. Lung volumes in CTs are ~10% of the scan volume - a not too unreasonable class balance. For this particular test application I've added the option of using NLLoss instead of the Dice Coefficient.

What does the PyTorch compute graph of Vnet look like?

You can see the compute graph here, which I created with make_graph.py, which I copied from densenet.pytorch which in turn was copied from Adam Paszke's gist.

Credits

The train.py script was derived from the one in the densenet.pytorch repo.

Note that the project description data, including the texts, logos, images, and/or trademarks, for each open source project belongs to its rightful owner. If you wish to add or remove any projects, please contact us at [email protected].