Neural Architecture Search of Semantic Segmentation Models (in PyTorch)
This repository provides official models from the paper Fast Neural Architecture Search of Compact Semantic Segmentation Models via Auxiliary Cells, available here
Fast Neural Architecture Search of Compact Semantic Segmentation Models via Auxiliary Cells
Vladimir Nekrasov, Hao Chen, Chunhua Shen, Ian Reid
To appear in CVPR, 2019
Getting Started
For flawless reproduction of our results, the Ubuntu OS is recommended. The models have been tested using Python 3.6.
Dependencies
pip3
Cython
cv2
jupyter-notebook
matplotlib
numpy
Pillow
torch>=1.0
torchvision
Inference Examples
For the ease of reproduction, we have embedded all our examples inside Jupyter notebooks.
Segmentation
Please refer to results on PASCAL VOC
Depth Estimation
Please refer to results on NYUD-v2
Search
If you would like to search for architectures yourself, please follow the instructions below:
Prepare data
You would need to have PASCAL VOC segmentation dataset expanded with annotations from BSD.
After that, run in your terminal:
mkdir data/datasets
ln -s /path_to_voc/VOCdevkit data/datasets/Running search
We rely on a Cython-based script for calculating mean IoU. In order to build the corresponding files, run the following:
cd src
python helpers/setup.py build_ext --build-lib=./helpers/After that, you can execute ./examples/search/search.sh that will start the search process.
[!] Please note that all the hyperparameters were tuned for running the search process on 2 GPUs, each with at least 11GB of memory. In case when your setup differs, you would need to tune the hyperparameters accordingly.
Checking the results
We output the log-file with all information on the search process. You can easily see top-performing architectures using the following command: python src/helpers/num_uq.py path_to_genotypes.out
License
This project is free to use for non-commercial purposes - see the LICENSE file for details.
Acknowledgments
- University of Adelaide and Australian Centre for Robotic Vision (ACRV) for making this project happen
- HPC Phoenix cluster at the University of Adelaide for making the training of the models possible
- PyTorch developers
- Yerba mate tea