sacmehta / Ynet
Licence: mit
Y-Net: Joint Segmentation and Classification for Diagnosis of Breast Biopsy Images
Stars: ✭ 100
Programming Languages
python
139335 projects - #7 most used programming language
Projects that are alternatives of or similar to Ynet
Bonnet
Bonnet: An Open-Source Training and Deployment Framework for Semantic Segmentation in Robotics.
Stars: ✭ 274 (+174%)
Mutual labels: convolutional-neural-networks, semantic-segmentation
Espnet
ESPNet: Efficient Spatial Pyramid of Dilated Convolutions for Semantic Segmentation
Stars: ✭ 473 (+373%)
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 (+189%)
Mutual labels: convolutional-neural-networks, semantic-segmentation
Self Driving Golf Cart
Be Driven 🚘
Stars: ✭ 147 (+47%)
Mutual labels: convolutional-neural-networks, semantic-segmentation
Espnetv2 Coreml
Semantic segmentation on iPhone using ESPNetv2
Stars: ✭ 66 (-34%)
Mutual labels: convolutional-neural-networks, semantic-segmentation
Bcdu Net
BCDU-Net : Medical Image Segmentation
Stars: ✭ 314 (+214%)
Mutual labels: convolutional-neural-networks, semantic-segmentation
Espnetv2
A light-weight, power efficient, and general purpose convolutional neural network
Stars: ✭ 377 (+277%)
Mutual labels: convolutional-neural-networks, semantic-segmentation
Vnet.pytorch
A PyTorch implementation for V-Net: Fully Convolutional Neural Networks for Volumetric Medical Image Segmentation
Stars: ✭ 506 (+406%)
Mutual labels: convolutional-neural-networks, semantic-segmentation
Minkowskiengine
Minkowski Engine is an auto-diff neural network library for high-dimensional sparse tensors
Stars: ✭ 1,110 (+1010%)
Mutual labels: convolutional-neural-networks, semantic-segmentation
Mtlnas
[CVPR 2020] MTL-NAS: Task-Agnostic Neural Architecture Search towards General-Purpose Multi-Task Learning
Stars: ✭ 58 (-42%)
Mutual labels: convolutional-neural-networks, semantic-segmentation
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 (-22%)
Mutual labels: convolutional-neural-networks, semantic-segmentation
Chainer Pspnet
PSPNet in Chainer
Stars: ✭ 76 (-24%)
Mutual labels: convolutional-neural-networks, semantic-segmentation
3dunet abdomen cascade
Stars: ✭ 91 (-9%)
Mutual labels: convolutional-neural-networks, semantic-segmentation
Cardiac Segmentation
Convolutional Neural Networks for Cardiac Segmentation
Stars: ✭ 94 (-6%)
Mutual labels: convolutional-neural-networks
Bayesian cnn
Bayes by Backprop implemented in a CNN in PyTorch
Stars: ✭ 98 (-2%)
Mutual labels: convolutional-neural-networks
Region Conv
Not All Pixels Are Equal: Difficulty-Aware Semantic Segmentation via Deep Layer Cascade
Stars: ✭ 95 (-5%)
Mutual labels: semantic-segmentation
Fakeimagedetector
Image Tampering Detection using ELA and CNN
Stars: ✭ 93 (-7%)
Mutual labels: convolutional-neural-networks
Lsd Seg
Learning from Synthetic Data: Addressing Domain Shift for Semantic Segmentation
Stars: ✭ 99 (-1%)
Mutual labels: semantic-segmentation
Har Keras Cnn
Human Activity Recognition (HAR) with 1D Convolutional Neural Network in Python and Keras
Stars: ✭ 97 (-3%)
Mutual labels: convolutional-neural-networks
Dped
Software and pre-trained models for automatic photo quality enhancement using Deep Convolutional Networks
Stars: ✭ 1,315 (+1215%)
Mutual labels: convolutional-neural-networks
Y-Net: Joint Segmentation and Classification for Diagnosis of Breast Biopsy Images
This repository contains the source code for our paper, YNet, which is accepted for publication at MICCAI'18.
Sample output of Y-Net
Y-Net identified correctly classified tissues that were not important for diagnosis. For example, stroma was identified as an important tissue, but blood was not. Stroma is an important tissue label for diagnosing breast cancer [1] and removing information about stroma decreased the diagnostic classification accuracy by about 4%. See paper for more details.
[1] Beck, Andrew H., et al. "Systematic analysis of breast cancer morphology uncovers stromal features associated with survival." Science translational medicine 3.108 (2011): 108ra113-108ra113.
Some segmentation results (Left: RGB WSI, Middle: Ground truth, Right: Predictions by Y-Net)
Structure of this repository
YNet is trained in two stages:
- stage1 This directory contains the source code for training the stage 1 in Y-Net. Stage 1 is nothing but a segmentation brach.
- stage2 This directory contains the source code for training the stage 2 in Y-Net. Stage 2 is jointly learning the segmentation and classification.
- seg_eval This directory contains the source code for producing the segmentation masks.
Pre-requisite
To run this code, you need to have following libraries:
- OpenCV - We tested our code with version 3.3.0. If you are using other versions, please change the source code accordingly.
- PyTorch - We tested with v0.2.0_4. If you are using other versions, please change the source code accordingly.
- Python - We tested our code with Python 3.6.2 (Anaconda custom 64-bit). If you are using other Python versions, please feel free to make necessary changes to the code.
We recommend to use Anaconda. We have tested our code on Ubuntu 16.04.
Citation
If Y-Net is useful for your research, then please cite our paper.
@inproceedings{mehta2018ynet,
title={{Y-Net: Joint Segmentation and Classification for Diagnosis of Breast Biopsy Images}},
author={Sachin Mehta and Ezgi Mercan and Jamen Bartlett and Donald Weaver and Joann Elmore and Linda Shapiro},
booktitle={International Conference on Medical image computing and computer-assisted intervention},
year={2018},
organization={Springer}
}
@article{mehta2018espnet,
title={ESPNet: Efficient Spatial Pyramid of Dilated Convolutions for Semantic Segmentation},
author={Sachin Mehta, Mohammad Rastegari, Anat Caspi, Linda Shapiro, and Hannaneh Hajishirzi},
journal={European Conference in Computer Vision (ECCV)},
year={2018}
}
License
This code is released under the same license terms as ESPNet.
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].