Thanks for your attention to the project.Now,the project can't reproduce the results.It just provides some ideas to reproduce the results.If you have some good ideas, both merging and discussing are welcome.
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