ENet Real Time Semantic Segmentation

ENet - Segmentación semántica en tiempo real

Una arquitectura de red neuronal para segmentación semántica en tiempo real.
En este repositorio hemos reproducido el documento ENet, que se puede utilizar en dispositivos móviles para la segmentación semántica en tiempo real. El enlace al artículo se puede encontrar aquí: ENet

¿Cómo utilizar?

0. Este repositorio viene con un práctico cuaderno que puedes usar con Colab.
   Puede encontrar un enlace al cuaderno aquí: ENet - Segmentación semántica en tiempo real
   Ábrelo en colab: Abrir en Colab

---

0. Clona el repositorio y cd en él.

 git clone https://github.com/iArunava/ENet-Real-Time-Semantic-Segmentation.git
cd ENet-Real-Time-Semantic-Segmentation/

1. Utilice este comando para entrenar el modelo.

 python3 init.py --mode train -iptr path/to/train/input/set/ -lptr /path/to/label/set/

2. Utilice este comando para probar el modelo.

 python3 init.py --mode test -m /path/to/the/pretrained/model.pth -i /path/to/image/to/infer.png

3. Utilice --help para obtener más comandos

 python3 init.py --help

Algunos resultados

Referencias

1. A. Paszke, A. Chaurasia, S. Kim y E. Culurciello. Enet: una arquitectura de red neuronal profunda para segmentación semántica en tiempo real. Preimpresión de arXiv arXiv:1606.02147, 2016.

Citas

 @inproceedings{ BrostowSFC:ECCV08,
  author    = {Gabriel J. Brostow and Jamie Shotton and Julien Fauqueur and Roberto Cipolla},
  title     = {Segmentation and Recognition Using Structure from Motion Point Clouds},
  booktitle = {ECCV (1)},
  year      = {2008},
  pages     = {44-57}
}

@article{ BrostowFC:PRL2008,
    author = "Gabriel J. Brostow and Julien Fauqueur and Roberto Cipolla",
    title = "Semantic Object Classes in Video: A High-Definition Ground Truth Database",
    journal = "Pattern Recognition Letters",
    volume = "xx",
    number = "x",   
    pages = "xx-xx",
    year = "2008"
}

Licencia

El código de este repositorio se distribuye bajo la licencia BSD v3.
Siéntete libre de bifurcar y disfrutar :)