All Projects → ndrplz → Dilation Tensorflow

ndrplz / Dilation Tensorflow

Licence: mit
A native Tensorflow implementation of semantic segmentation according to Multi-Scale Context Aggregation by Dilated Convolutions (2016). Optionally uses the pretrained weights by the authors.

Programming Languages

python
139335 projects - #7 most used programming language

Projects that are alternatives of or similar to Dilation Tensorflow

Seg Mentor
TFslim based semantic segmentation models, modular&extensible boutique design
Stars: ✭ 43 (-67.91%)
Mutual labels:  segmentation, semantic-segmentation
Vision4j Collection
Collection of computer vision models, ready to be included in a JVM project
Stars: ✭ 132 (-1.49%)
Mutual labels:  segmentation, semantic-segmentation
Jacinto Ai Devkit
Training & Quantization of embedded friendly Deep Learning / Machine Learning / Computer Vision models
Stars: ✭ 49 (-63.43%)
Mutual labels:  segmentation, semantic-segmentation
Lightnet
LightNet: Light-weight Networks for Semantic Image Segmentation (Cityscapes and Mapillary Vistas Dataset)
Stars: ✭ 698 (+420.9%)
Mutual labels:  segmentation, semantic-segmentation
Setr Pytorch
Rethinking Semantic Segmentation from a Sequence-to-Sequence Perspective with Transformers
Stars: ✭ 96 (-28.36%)
Mutual labels:  segmentation, semantic-segmentation
Medicaldetectiontoolkit
The Medical Detection Toolkit contains 2D + 3D implementations of prevalent object detectors such as Mask R-CNN, Retina Net, Retina U-Net, as well as a training and inference framework focused on dealing with medical images.
Stars: ✭ 917 (+584.33%)
Mutual labels:  segmentation, semantic-segmentation
Semanticsegmentation
A framework for training segmentation models in pytorch on labelme annotations with pretrained examples of skin, cat, and pizza topping segmentation
Stars: ✭ 52 (-61.19%)
Mutual labels:  segmentation, semantic-segmentation
Erfnet pytorch
Pytorch code for semantic segmentation using ERFNet
Stars: ✭ 304 (+126.87%)
Mutual labels:  segmentation, semantic-segmentation
3dunet abdomen cascade
Stars: ✭ 91 (-32.09%)
Mutual labels:  segmentation, semantic-segmentation
Deep Segmentation
CNNs for semantic segmentation using Keras library
Stars: ✭ 69 (-48.51%)
Mutual labels:  segmentation, semantic-segmentation
Efficient Segmentation Networks
Lightweight models for real-time semantic segmentationon PyTorch (include SQNet, LinkNet, SegNet, UNet, ENet, ERFNet, EDANet, ESPNet, ESPNetv2, LEDNet, ESNet, FSSNet, CGNet, DABNet, Fast-SCNN, ContextNet, FPENet, etc.)
Stars: ✭ 579 (+332.09%)
Mutual labels:  segmentation, semantic-segmentation
Crfasrnn pytorch
CRF-RNN PyTorch version http://crfasrnn.torr.vision
Stars: ✭ 102 (-23.88%)
Mutual labels:  segmentation, semantic-segmentation
Superpoint graph
Large-scale Point Cloud Semantic Segmentation with Superpoint Graphs
Stars: ✭ 533 (+297.76%)
Mutual labels:  segmentation, semantic-segmentation
Deeplabv3 Plus
Tensorflow 2.3.0 implementation of DeepLabV3-Plus
Stars: ✭ 32 (-76.12%)
Mutual labels:  segmentation, semantic-segmentation
Bcdu Net
BCDU-Net : Medical Image Segmentation
Stars: ✭ 314 (+134.33%)
Mutual labels:  segmentation, semantic-segmentation
Ccnet
CCNet: Criss-Cross Attention for Semantic Segmentation (TPAMI 2020 & ICCV 2019).
Stars: ✭ 1,059 (+690.3%)
Mutual labels:  segmentation, semantic-segmentation
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 (+120.9%)
Mutual labels:  segmentation, semantic-segmentation
Segmentation models.pytorch
Segmentation models with pretrained backbones. PyTorch.
Stars: ✭ 4,584 (+3320.9%)
Mutual labels:  segmentation, semantic-segmentation
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 (-50%)
Mutual labels:  segmentation, semantic-segmentation
Segmentation
Tensorflow implementation : U-net and FCN with global convolution
Stars: ✭ 101 (-24.63%)
Mutual labels:  segmentation, semantic-segmentation

dilation-tensorflow

A native Tensorflow implementation of semantic segmentation according to Multi-Scale Context Aggregation by Dilated Convolutions by Yu and Koltun.

Pretrained weights have been converted to TensorFlow from the original Caffe implementation.

Model pretrained either on CityScapes or on CamVid datasets is available.

You you're looking instead for a Keras+Theano implementation of this very same network you can find it here.

Examples

Cityscapes

input
Test image (input)

segmentation
Test image (prediction)

CamVid

input
Test image (input)

segmentation
Test image (prediction)

How-to

  1. Download pretrained weights from here:

    CityScapes weights

    CamVid weights

  2. Move weights file into data directory.

  3. Run the model on the test image by executing main_tf.py.

Configuration

This model has been tested with the following configuration:

  • Ubuntu 16.04
  • python 3.5.2
  • tensorflow 1.1.0
  • cv2 3.2.0

Acknowledgements

Thanks to DavideA which converted all weights from Caffe to Keras+Theano (here) making my effort of conversion towards TensorFlow much less painful than it could have been :-)

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