/AutoDeeplab

Pytorch Implementation the paper Auto-DeepLab Hierarchical Neural Architecture Search for Semantic Image Segmentation

Primary LanguagePython

Statement

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.

AutoDeeplab

This is an implementation of Auto-DeepLab using Pytorch.

Environment

The implementation needs the following dependencies:

  • Python = 3.7

  • Pytorch = 0.4

  • TensorboardX

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

Installation

First, clone the repository

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

Then

cd AutoDeeplab

Train

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

Reference

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

[2] : pytorch-deeplab-xception