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 → shreyaspadhy → Unet Zoo

shreyaspadhy / Unet Zoo

Licence: mit
A collection of UNet and hybrid architectures in PyTorch for 2D and 3D Biomedical Image segmentation

Programming Languages

python
139335 projects - #7 most used programming language

Labels

deep-learningpytorchcnnlstmrnnunet

Projects that are alternatives of or similar to Unet Zoo

Deepseqslam
The Official Deep Learning Framework for Route-based Place Recognition
Stars: ✭ 49 (-83.77%)
Mutual labels:  cnn, lstm, rnn
Eeg Dl
A Deep Learning library for EEG Tasks (Signals) Classification, based on TensorFlow.
Stars: ✭ 165 (-45.36%)
Mutual labels:  cnn, lstm, rnn
Cnn lstm for text classify
CNN, LSTM, NBOW, fasttext 中文文本分类
Stars: ✭ 90 (-70.2%)
Mutual labels:  cnn, lstm, rnn
Deep Music Genre Classification
🎵 Using Deep Learning to Categorize Music as Time Progresses Through Spectrogram Analysis
Stars: ✭ 23 (-92.38%)
Mutual labels:  cnn, lstm, rnn
Pytorch Sentiment Analysis
Tutorials on getting started with PyTorch and TorchText for sentiment analysis.
Stars: ✭ 3,209 (+962.58%)
Mutual labels:  cnn, lstm, rnn
Rnn Theano
使用Theano实现的一些RNN代码，包括最基本的RNN，LSTM，以及部分Attention模型，如论文MLSTM等
Stars: ✭ 31 (-89.74%)
Mutual labels:  cnn, lstm, rnn
Pytorch Learners Tutorial
PyTorch tutorial for learners
Stars: ✭ 97 (-67.88%)
Mutual labels:  cnn, lstm, rnn
Pytorch Pos Tagging
A tutorial on how to implement models for part-of-speech tagging using PyTorch and TorchText.
Stars: ✭ 96 (-68.21%)
Mutual labels:  cnn, lstm, rnn
Natural Language Processing With Tensorflow
Natural Language Processing with TensorFlow, published by Packt
Stars: ✭ 222 (-26.49%)
Mutual labels:  cnn, lstm, rnn
Sign Language Gesture Recognition
Sign Language Gesture Recognition From Video Sequences Using RNN And CNN
Stars: ✭ 214 (-29.14%)
Mutual labels:  cnn, lstm, rnn
Multi Class Text Classification Cnn Rnn
Classify Kaggle San Francisco Crime Description into 39 classes. Build the model with CNN, RNN (GRU and LSTM) and Word Embeddings on Tensorflow.
Stars: ✭ 570 (+88.74%)
Mutual labels:  cnn, lstm, rnn
Lightnet
Efficient, transparent deep learning in hundreds of lines of code.
Stars: ✭ 243 (-19.54%)
Mutual labels:  cnn, lstm, rnn
Video Classification
Tutorial for video classification/ action recognition using 3D CNN/ CNN+RNN on UCF101
Stars: ✭ 543 (+79.8%)
Mutual labels:  cnn, lstm, rnn
Neural Networks
All about Neural Networks!
Stars: ✭ 34 (-88.74%)
Mutual labels:  cnn, lstm, rnn
Basicocr
BasicOCR是一个致力于解决自然场景文字识别算法研究的项目。该项目由长城数字大数据应用技术研究院佟派AI团队发起和维护。
Stars: ✭ 336 (+11.26%)
Mutual labels:  cnn, lstm, rnn
Keraspp
코딩셰프의 3분 딥러닝, 케라스맛
Stars: ✭ 178 (-41.06%)
Mutual labels:  cnn, rnn, unet
Caption generator
A modular library built on top of Keras and TensorFlow to generate a caption in natural language for any input image.
Stars: ✭ 243 (-19.54%)
Mutual labels:  cnn, lstm, rnn
Automatic speech recognition
End-to-end Automatic Speech Recognition for Madarian and English in Tensorflow
Stars: ✭ 2,751 (+810.93%)
Mutual labels:  cnn, lstm, rnn
totally humans
rnn trained on r/totallynotrobots 🤖
Stars: ✭ 23 (-92.38%)
Mutual labels:  lstm, rnn
time-series-forecasting-tensorflowjs
Pull stock prices from online API and perform predictions using Long Short Term Memory (LSTM) with TensorFlow.js framework
Stars: ✭ 96 (-68.21%)
Mutual labels:  lstm, rnn
View All Similar Projects ➔

UNet-Zoo

A collection of UNet and hybrid architectures for 2D and 3D Biomedical Image segmentation, implemented in PyTorch.

This repository contains a collection of architectures used for Biomedical Image Segmentation, implemented on the BraTS Brain Tumor Segmentation Challenge Dataset. The following architectures are implemented

  1. UNet - Standard UNet architecture as described in the Ronneberger et al 2015 paper [reference]

  1. Small UNet - 40x smaller version of UNet that achieves similar performance [Theano Implementation]

  1. UNet with BDCLSTM - Combining a BDC-LSTM network with UNet to encode spatial correlation for 3D segmentation [reference]

  1. kUNet - Combining multiple UNets for increasing heirarchial preservation of information (coming soon) [reference]
  2. R-UNet - UNet with recurrent connections for another way to encode $z$-context (coming soon)

To Run

First, apply for access the BraTS Tumor dataset, and place the scans in a Data/ folder, divided into Train and Test. Feel free to modify the BraTS PyTorch dataloaders in data.py for your use.

  1. UNet - run main.py, type --help for information on arguments. Example: python main.py --train --cuda --data-folder "./Data/"
  2. Small UNet - run main_small.py, and use --help
  3. BDC-LSTM - run main_bdclstm.py and use the weights for either your trained UNet or Small-UNet models (--help is your savior).

Some Results

  1. Comparisons of UNet (top) and Small UNet (bottom)

  1. DICE Scores for UNet and Small UNet

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