This repo is the pytorch implementation of paper Colorful Image Colorization. This implementation is sightly different than the paper.
- pytorch 0.3+ and torchvision 0.1
- python 3.5
- Basic packages (numpy, scipy, matplotlib, tensorboardx)
Start from scratch
python main.py --batch_size 32 --gpu 0,1 --num_iterations 40000 --save_directory <path/to save weights>
Or Resume from a checkpoint
python main.py --resume <iteration_number>
python main.py --infer_iter <which iteration weight to pick for inference> --weight_dir <path to saved weight>