神经网络python识别语义_ENet - 一种用于实时语义分割的神经网络体系结构

ENet - Real Time Semantic Segmentation

A Neural Net Architecture for real time Semantic Segmentation.

In this repository we have reproduced the ENet Paper - Which can be used on mobile devices for real time semantic segmentattion. The link to the paper can be found here: ENet

How to use?

This repository comes in with a handy notebook which you can use with Colab.

You can find a link to the notebook here: ENet - Real Time Semantic Segmentation

Open it in colab: Open in Colab

Clone the repository and cd into it

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

cd ENet-Real-Time-Semantic-Segmentation/

Use this command to train the model

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

Use this command to test the model

python3 init.py --mode test -m /path/to/the/pretrained/m