参考图像彩色化网络修改流程（自用版）

最新推荐文章于 2024-08-02 23:48:48 发布

钰宸y

最新推荐文章于 2024-08-02 23:48:48 发布

阅读量100

点赞数

分类专栏：彩色化问题人工智能论文文章标签：计算机视觉 pytorch 人工智能机器学习深度学习

本文链接：https://blog.csdn.net/qq_40721108/article/details/133998371

版权

论文同时被 3 个专栏收录

19 篇文章 6 订阅

订阅专栏

彩色化问题

9 篇文章 0 订阅

订阅专栏

人工智能

8 篇文章 2 订阅

订阅专栏

一. SGA-Eliminating Gradient Conflict in Reference-based Line-Art Colorization（2022ECCV）

1. 修改config.yml
修改前

  EPOCH: 40
  BATCH_SIZE: 16
  NUM_WORKER : 4
  TRAIN_DIR : 'anime' # 'anime' or 'afhq_cat' or 'afhq_dog' or afhq_wild

修改后

  EPOCH: 400
  BATCH_SIZE: 8
  NUM_WORKER : 0
  TRAIN_DIR : 'nighttime' # 'anime' or 'afhq_cat' or 'afhq_dog' or afhq_wild

添加

USE_TENSORBOARD : 'True'

2. 修改data_loader.py
添加

 elif config['TRAINING_CONFIG']['TRAIN_DIR'] == 'nighttime':
     self.img_dir = r'F:\RefDataset\KAIST\train\refB'
     self.skt_dir = r'F:\RefDataset\KAIST\train\nightA'
     self.data_list = glob.glob(os.path.join(self.img_dir, '*.jpg'))

3. 修改model.py

self.gcn3 = Gconv(in_features=channel, out_features=channel)
self.gcn4 = Gconv(in_features=channel, out_features=channel)

修改为

self.gcn3 = Gconv(in_ch=channel, out_ch=channel)
self.gcn4 = Gconv(in_ch=channel, out_ch=channel)

二. SCFT-Reference-Based Sketch Image Colorization using Augmented-Self Reference and Dense Semantic Correspondence（2020CVPR）

和SGA修改一致
1. 修改config.yml
修改前

  EPOCH: 40
  BATCH_SIZE: 16
  NUM_WORKER : 4
  TRAIN_DIR : 'anime' # 'anime' or 'afhq_cat' or 'afhq_dog' or afhq_wild

修改后

  EPOCH: 400
  BATCH_SIZE: 8
  NUM_WORKER : 0
  TRAIN_DIR : 'nighttime' # 'anime' or 'afhq_cat' or 'afhq_dog' or afhq_wild

添加

USE_TENSORBOARD : 'True'

2. 修改data_loader.py
添加

 elif config['TRAINING_CONFIG']['TRAIN_DIR'] == 'nighttime':
     self.img_dir = r'F:\RefDataset\KAIST\train\refB'
     self.skt_dir = r'F:\RefDataset\KAIST\train\nightA'
     self.data_list = glob.glob(os.path.join(self.img_dir, '*.jpg'))

三.SPADE-Semantic Image Synthesis with Spatially-Adaptive Normalization（2019CVPR）

下载Synchronized-BatchNorm-PyTorch

https://github.com/vacancy/Synchronized-BatchNorm-PyTorch

下载后放到./models/networks/

GitHub提供训练和测试代码

python train.py --name [experiment_name] --dataset_mode custom --label_dir [path_to_labels] -- image_dir [path_to_images] --label_nc [num_labels]
python test.py --name [name_of_experiment] --dataset_mode [dataset_mode] --dataroot [path_to_dataset]

python train.py --name nighttime --dataset_mode custom --label_dir F:\RefDataset\KAIST\train\nightA -- image_dir F:\RefDataset\KAIST\train\refB --label_nc 182

问题1：

IndexError: list index out of range

更改

python train.py --name nighttime --dataset_mode custom --label_dir F:\RefDataset\KAIST\train\nightA --image_dir F:\RefDataset\KAIST\train\refB --label_nc 182 --no_instance

问题2：

RuntimeError: cuDNN error: CUDNN_STATUS_INTERNAL_ERROR

更改

python train.py --name nighttime --dataset_mode custom --dataroot F:\RefDataset\iRay\train --label_dir F:\RefDataset\iRay\train\nightA --image_dir F:\RefDataset\iRay\train\refB --no_instance --contain_dontcare_label --label_nc 512 --no_flip --preprocess_mode none --load_size 256 --crop_size 256 --display_freq 5000 --save_epoch_freq 100 --niter 100--gpu_ids 0
python test.py --name nighttime --dataset_mode custom --dataroot /home/sys120-1/cy/ref_based/iRay/train  --label_dir /home/sys120-1/cy/ref_based/iRay/train/nightA --image_dir /home/sys120-1/cy/ref_based/iRay/train/refB --no_instance --contain_dontcare_label --label_nc 512 --no_flip --preprocess_mode none --load_size 256 --crop_size 256 --gpu_ids 0