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