Semantic Segmentation DeconvNet PyTorch
Data:
- PASCAL VOC 2012 segmentation dataset (train + val)
- Download http://host.robots.ox.ac.uk/pascal/VOC/voc2012/VOCtrainval_11-May-2012.tar
- extract to ./data
Data Preprocessing:
color map (3 * 224 * 224) --> one-hot class map (21 * 224 * 224)
DeconvNet training:
- epochs = 50
- Batch size = 64
- Learning rate = 0.001
- Loss func = CrossEntropyLoss