This is an implementation of Auto-DeepLab using Pytorch.

The implementation needs the following dependencies:

• Python = 3.7

• Pytorch = 0.4

• TensorboardX

Other basic dependencies like matplotlib, tqdm ... are also needed.

First, clone the repository

git clone https://github.com/MenghaoGuo/AutoDeeplab.git

Then

cd AutoDeeplab

The dataloader module is built on this repo

If you want to train this model on different datasets, you need to edit --dataset parameter and then:

bash train_voc.sh

[1] : Auto-DeepLab: Hierarchical Neural Architecture Search for Semantic Image Segmentation

[2] : pytorch-deeplab-xception