Colorization problem

[songyn95]

Hello, I want to convert infrared images at night into RGB. It is required that the effect of RGB should be consistent with that during the day.

• I see that the author has many colorization methods (pix2pix, interactive deep colorization). Which method is the most suitable?
• In addition, if I use pix2pix method, whether the following color training and test commands are correct:
  1. python train.py --dataroot ./single/colorization_IR --name IR_colorization --model colorization --load_size 2560 --preprocess crop --crop_size 640 --batch_size 2 --netG resnet_9blocks --gpu_ids 0,1,2,3,4,5,6,7
  2. python test.py --dataroot ./single/colorization_IR/ --name IR_colorization --model colorization --dataset_mode colorization --preprocess none --netG resnet_9blocks

Look forward to your reply！