abinjoabraham / AutoML

AutoDeeplab / auto-deeplab implemented in Pytorch

AutoML for Image Segmentation and Detection

This is an open-source project of AutoML for object detection & segmentation as well as semantic segmentation.

Currently this repo contains a pytorch implementation for AutoDeeplab.

Requirements

• Pytorch version 1.1

• Python 3

• tensorboardX

• torchvision

• pycocotools

• tqdm

• numpy

• pandas

Training

###Start training

CUDA_VISIBLE_DEVICES=0 python train_autodeeplab.py --dataset cityscapes

###Resume training

CUDA_VISIBLE_DEVICES=0 python train_autodeeplab.py --dataset cityscapes --resume /AutoDeeplabpath/checkpoint.pth.tar

References

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

[2] : pytorch-deeplab-xception

[3] : Some code for the project was taken from here