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