Semantic-aware automatic image colorization via unpaired cycle-consistent self-supervised network
Semantic-aware automatic image colorization via unpaired cycle-consistent self-supervised network
复制标题
通过不成对的循环实现语义感知的自动图像着色——一致的自监督网络
DOI:
10.1002/int.22667
复制
发表时间:
2021-09-19
影响因子:
7
通讯作者:
Wang, Mingwen
中科院分区:
文献类型:
--
作者:
Xiao, Yuxuan;Jiang, Aiwen;Wang, Mingwen
Automatic image colorization without manual interventions is an ill-conditioned and inherently ambiguous problem. Most of existing methods focus on formulating colorization as a regression problem and learn parametric mappings from grayscale to color through deep neural networks. Due to the multimodalities of color-grayscale space, in many applications, it is not required to recover exact ground-truth color. Pair-wise pixel-to-pixel learning-based algorithms lack rationality. Techniques such as color space conversion techniques are then proposed to avoid such direct pixel learning. However, the coloring results after color space conversion are blunt and unnatural. In this paper, we hold viewpoints that a reasonable solution is to generate some colorized result that looks natural. No matter what color a region is to be assigned, the colorized region should be semantically and spatially consistent. In this paper, we propose an effective semantic-aware automatic colorization model via unpaired cycle-consistent self-supervised network. Low-level monochrome loss, perceptual identity loss and high-level semantic-consistence loss, together with adversarial loss, are introduced to guide network self-training. We train and test our model on randomly selected subsets from PASCAL VOC 2012. The experimental results including human subjective studies demonstrate that, compared with state-of-the-art methods, our proposed model can achieve more convincing and superior results. Relevant source code is available at .