ENet Real Time Semantic Segmentation

ENet – Semantische Segmentierung in Echtzeit

Eine neuronale Netzarchitektur für die semantische Segmentierung in Echtzeit.
In diesem Repository haben wir das ENet-Papier reproduziert, das auf mobilen Geräten für die semantische Segmentierung in Echtzeit verwendet werden kann. Den Link zum Artikel finden Sie hier: ENet

Wie benutzt man?

0. Zu diesem Repository gehört ein praktisches Notizbuch, das Sie mit Colab verwenden können.
   Einen Link zum Notebook finden Sie hier: ENet – Real Time Semantic Segmentation
   In Colab öffnen: In Colab öffnen

---

0. Klonen Sie das Repository und cd hinein

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

1. Verwenden Sie diesen Befehl, um das Modell zu trainieren

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

2. Verwenden Sie diesen Befehl, um das Modell zu testen

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

3. Verwenden Sie --help um weitere Befehle zu erhalten

 python3 init.py --help

Einige Ergebnisse

Referenzen

1. A. Paszke, A. Chaurasia, S. Kim und E. Culurciello. Enet: Eine tiefe neuronale Netzwerkarchitektur für die semantische Segmentierung in Echtzeit. arXiv-Vorabdruck arXiv:1606.02147, 2016.

Zitate

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

Lizenz

Der Code in diesem Repository wird unter der BSD v3-Lizenz vertrieben.
Fühlen Sie sich frei, sich zu teilen und zu genießen :)