DeepLab with PyTorch
Unofficial implementation to train DeepLab v2 (ResNet-101) on COCO-Stuff 10k dataset. DeepLab is one of the CNN architectures for semantic image segmentation. COCO Stuff 10k is a semantic segmentation dataset, which includes 10k images from 182 thing/stuff classes.
Requirements
pytorch
pytorch 0.4.0
torchvision
cv2 >= 3.0.0
tqdm
click
addict
h5py (for COCO-Stuff v1.0)
scipy (for COCO-Stuff v1.1)
Usage
Preparation
Dataset
Download COCO-Stuff 10k dataset and unzip it.
Set the path to the dataset in config/cocostuff.yaml.
cocostuff-10k-v1.1
├── images
│ ├── COCO_train2014_000000000077.jpg
│ └── ...
├── annotations
│ ├── COCO_train2014_000000000077.mat
│ └── ...
└── imageLists
├── all.txt
├── test.