Skip to content

Instantly share code, notes, and snippets.

View jeanpat's full-sized avatar

Pommier jeanpat

View GitHub Profile
@jeanpat
jeanpat / model.py
Created March 19, 2017 10:21 — forked from ndronen/model.py
Semantic segmentation with ENet in PyTorch
#!/usr/bin/env python
"""
A quick, partial implementation of ENet (https://arxiv.org/abs/1606.02147) using PyTorch.
The original Torch ENet implementation can process a 480x360 image in ~12 ms (on a P2 AWS
instance). TensorFlow takes ~35 ms. The PyTorch implementation takes ~25 ms, an improvement
over TensorFlow, but worse than the original Torch.
"""
from __future__ import absolute_import