All Projects → imlab-uiip → Lung Segmentation 2d

imlab-uiip / Lung Segmentation 2d

Licence: mit
Lung fields segmentation on CXR images using convolutional neural networks.

Programming Languages

python
139335 projects - #7 most used programming language

Projects that are alternatives of or similar to Lung Segmentation 2d

Data Science Bowl 2018
End-to-end one-class instance segmentation based on U-Net architecture for Data Science Bowl 2018 in Kaggle
Stars: ✭ 56 (-59.42%)
Mutual labels:  convolutional-neural-networks, segmentation, unet
Pytorch Saltnet
Kaggle | 9th place single model solution for TGS Salt Identification Challenge
Stars: ✭ 270 (+95.65%)
Mutual labels:  convolutional-neural-networks, segmentation, unet
Bcdu Net
BCDU-Net : Medical Image Segmentation
Stars: ✭ 314 (+127.54%)
Mutual labels:  convolutional-neural-networks, segmentation, unet
Unet 3d
3D Unet Equipped with Advanced Deep Learning Methods
Stars: ✭ 57 (-58.7%)
Mutual labels:  segmentation, unet
Meshcnn
Convolutional Neural Network for 3D meshes in PyTorch
Stars: ✭ 1,032 (+647.83%)
Mutual labels:  convolutional-neural-networks, segmentation
Switchnorm segmentation
Switchable Normalization for semantic image segmentation and scene parsing.
Stars: ✭ 47 (-65.94%)
Mutual labels:  convolutional-neural-networks, segmentation
Medicalzoopytorch
A pytorch-based deep learning framework for multi-modal 2D/3D medical image segmentation
Stars: ✭ 546 (+295.65%)
Mutual labels:  segmentation, unet
Quicknat pytorch
PyTorch Implementation of QuickNAT and Bayesian QuickNAT, a fast brain MRI segmentation framework with segmentation Quality control using structure-wise uncertainty
Stars: ✭ 74 (-46.38%)
Mutual labels:  convolutional-neural-networks, segmentation
Pointcnn
PointCNN: Convolution On X-Transformed Points (NeurIPS 2018)
Stars: ✭ 1,120 (+711.59%)
Mutual labels:  convolutional-neural-networks, segmentation
Dlcv for beginners
《深度学习与计算机视觉》配套代码
Stars: ✭ 1,244 (+801.45%)
Mutual labels:  segmentation, unet
3dunet abdomen cascade
Stars: ✭ 91 (-34.06%)
Mutual labels:  convolutional-neural-networks, segmentation
Segmentation Networks Benchmark
Evaluation framework for testing segmentation networks in Keras
Stars: ✭ 34 (-75.36%)
Mutual labels:  segmentation, unet
Torchio
Medical image preprocessing and augmentation toolkit for deep learning
Stars: ✭ 708 (+413.04%)
Mutual labels:  convolutional-neural-networks, segmentation
Niftynet
[unmaintained] An open-source convolutional neural networks platform for research in medical image analysis and image-guided therapy
Stars: ✭ 1,276 (+824.64%)
Mutual labels:  convolutional-neural-networks, segmentation
Segmentation
Tensorflow implementation : U-net and FCN with global convolution
Stars: ✭ 101 (-26.81%)
Mutual labels:  segmentation, unet
Unet Segmentation Pytorch Nest Of Unets
Implementation of different kinds of Unet Models for Image Segmentation - Unet , RCNN-Unet, Attention Unet, RCNN-Attention Unet, Nested Unet
Stars: ✭ 683 (+394.93%)
Mutual labels:  segmentation, unet
Multiclass Semantic Segmentation Camvid
Tensorflow 2 implementation of complete pipeline for multiclass image semantic segmentation using UNet, SegNet and FCN32 architectures on Cambridge-driving Labeled Video Database (CamVid) dataset.
Stars: ✭ 67 (-51.45%)
Mutual labels:  segmentation, unet
Unet Family
Paper and implementation of UNet-related model.
Stars: ✭ 1,924 (+1294.2%)
Mutual labels:  segmentation, unet
Unet
unet for image segmentation
Stars: ✭ 3,751 (+2618.12%)
Mutual labels:  segmentation, unet
Brats17
Patch-based 3D U-Net for brain tumor segmentation
Stars: ✭ 85 (-38.41%)
Mutual labels:  segmentation, unet

Lung Segmentation (2D)

Repository features UNet inspired architecture used for segmenting lungs on chest X-Ray images.

Demo

See the application of the model in Demo folder.

Implementation

Implemented in Keras(2.0.4) with TensorFlow(1.1.0) as backend.

Use of data augmentation for training required slight changes to keras ImageDataGenerator. Generator in image_gen.py applies same transformation to both the image and the label mask.

To use this implementation one needs to load and preprocess data (see load_data.py), train new model if needed (train_model.py) and use the model for generating lung masks (inference.py).

trained_model.hdf5 contains model trained on both data sets mentioned below.

Segmentation

Scores achieved on Montgomery and JSRT(With these masks. See preprocess_JSRT.py.) (Measured using 5-fold cross-validation):

JSRT Montgomery
IoU 0.971 0.956
Dice 0.985 0.972

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].