DSSLIC
Pytorch implmementation of DSSLIC: Deep Semantic Segmentation-based Layered Image Compression
Paper: https://arxiv.org/abs/1806.03348
Requirements:
- Ubuntu 16.04
- Python 2.7
- Cuda 8.0
- Pyorch 0.3.0
Training
ADE20K Dataset
python train.py --name ADE20K_model --dataroot ./datasets/ADE20K/ --label_nc 151 --loadSize 256 --resize_or_crop resize --batchSize 8
Cityscapes Dataset
python train.py --name cityscapes_model --dataroot ./datasets/cityscapes/ --label_nc 35 --loadSize 1024 --resize_or_crop scale_width --batchSize 2
Testing
ADE20K testset
python test.py --name ADE20K_model --dataroot ./datasets/ADE20K/ --label_nc 151 --resize_or_crop none --batchSize 1 --how_many 50
Kodak testset
python test.py --name ADE20K_model --dataroot ./datasets/Kodak/ --label_nc 151 --resize_or_crop none --batchSize 1 --how_many 24
Cityscapes testset
python test.py --name Cityscapes_model --dataroot ./datasets/cityscapes/ --label_nc 35 --loadSize 1024 --resize_or_crop scale_width --batchSize 1 --how_many 50