ndrplz/dilation-tensorflow

Stars
136
Rank 267,054 (Top 6 %)
Language
Python
License
MIT License
Created over 7 years ago
Updated over 6 years ago

ndrplz/dilation-tensorflow

ndrplz

There are no reviews yet. Be the first to send feedback to the community and the maintainers!

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.

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

Test image (input)	Test image (prediction)

CamVid

Test image (input)	Test image (prediction)

How-to

Download pretrained weights from here:

CityScapes weights

CamVid weights
Move weights file into data directory.
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 :-)

self-driving-car

Udacity Self-Driving Car Engineer Nanodegree projects.

ConvLSTM_pytorch

Implementation of Convolutional LSTM in PyTorch.

google-drive-downloader

Minimal class to download shared files from Google Drive.

surround_vehicles_awareness

Learn to map surrounding vehicles onto a bird's eye view of the scene.

machine_learning_lectures

Collection of lectures and lab lectures on machine learning and deep learning. Lab practices in Python and TensorFlow.

dreyeve

[TPAMI 2018] Predicting the Driver’s Focus of Attention: the DR(eye)VE Project. A deep neural network learnt to reproduce the human driver focus of attention (FoA) in a variety of real-world driving scenarios.

differentiable-renderer

Rastering algorithm to approximate the rendering of a 3D model silhouette in a fully differentiable way.

small_norb

Python wrapper to small NORB dataset

semiparametric

[TPAMI 2020] Generating Novel Views of Vehicles via Semi-parametric Guidance. A semi-parametric approach for synthesizing novel views of a rigid object from a single monocular image.

planar-distance-estimation

This repository accompanies the laboratory practice on Planar Distance Estimation for the AI4Automotive course at University of Modena and Reggio Emilia.

transforming-autoencoders

Transforming Autoencoder (Hinton et al.) implementation in TensorFlow. A way to get hands dirty with Hinton's capsules.

computer_vision_utils

Everything that I code more than twice during my PhD will end up here.

CIFAR-10

Python plug-and-play wrapper to CIFAR-10 dataset.

eurlex-toolbox

Python toolbox to load, parse and process Official Journals of the European Union (EU).

haralick-labeling-visualized

Visual explanation and python implementation of Haralick algorithm for connected component labeling

lesser-known-python

Collection of lesser-known python features that you will love (and use).

Jupyter Notebook

micromachine

toy example of q-learning algorithm

dotfiles

just my dotfiles

pong_motion

Pong game based on motion tracking of the two players.