ENet Real Time Semantic Segmentation

ENet - Segmentation sémantique en temps réel

Une architecture de réseau neuronal pour la segmentation sémantique en temps réel.
Dans ce référentiel, nous avons reproduit l'ENet Paper - qui peut être utilisé sur les appareils mobiles pour une segmentation sémantique en temps réel. Le lien vers l'article peut être trouvé ici : ENet

Comment utiliser ?

0. Ce référentiel est livré avec un bloc-notes pratique que vous pouvez utiliser avec Colab.
   Vous pouvez trouver un lien vers le notebook ici : ENet - Segmentation sémantique en temps réel
   Ouvrir dans Colab : Ouvrir dans Colab

---

0. Clonez le référentiel et insérez-y un CD

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

1. Utilisez cette commande pour entraîner le modèle

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

2. Utilisez cette commande pour tester le modèle

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

3. Utilisez --help pour obtenir plus de commandes

 python3 init.py --help

Quelques résultats

Références

1. A. Paszke, A. Chaurasia, S. Kim et E. Culurciello. Enet : Une architecture de réseau neuronal profonde pour la segmentation sémantique en temps réel. Préimpression arXiv arXiv : 1606.02147, 2016.

Citations

 @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"
}

Licence

Le code de ce référentiel est distribué sous la licence BSD v3.
N'hésitez pas à bifurquer et à savourer :)