Dou Zhi, Wang Ning, Li Baopu, Wang Zhihui, Li Haojie, Liu Bin
IEEE Trans Image Process. 2021;30:7292-7304. doi: 10.1109/TIP.2021.3104190. Epub 2021 Aug 20.
Automatic sketch colorization is a challenging task in both computer graphics and computer vision since all the color, texture, shading generation have to be created based on the abstract sketch. Besides, it is a subjective task in painting process, which needs illustrators to comprehend drawing priori (DP), such as hue variation, saturation contrast and gray contrast and utilize them in the HSV color space which is closer to human visual cognition system. As such, incorporating supplementary supervision in the HSV color space may be beneficial to sketch colorization. However, previous methods improve the colorization quality only in the RGB color space without considering the HSV color space, often causing results with dull color, inappropriate saturation contrast, and artifacts. To address this issue, we propose a novel sketch colorization method, dual color space guided generative adversarial network (DCSGAN), that considers the complementary information contained in both the RGB and HSV color space. Specifically, we incorporate the HSV color space to construct dual color spaces for supervising our method with a color space transformation (CST) network that learns transformation from the RGB to HSV color space. Then, we propose a DP loss that enables the DCSGAN to generate vivid color images with pixel level supervision. Additionally, a novel dual color space adversarial (DCSA) loss is designed to guide the generator at global level to reduce the artifacts to meet audiences' aesthetic expectations. Extensive experiments and ablation studies demonstrate the superiority of the proposed method over previous state-of-the-art (SOTA) methods.
自动草图上色在计算机图形学和计算机视觉领域都是一项具有挑战性的任务,因为所有颜色、纹理、阴影的生成都必须基于抽象的草图来创建。此外,这在绘画过程中是一项主观性任务,需要绘图师理解绘画先验知识(DP),如色调变化、饱和度对比和灰度对比,并在更接近人类视觉认知系统的HSV颜色空间中加以运用。因此,在HSV颜色空间中引入补充监督可能有助于草图上色。然而,以往的方法仅在RGB颜色空间中提高上色质量,而未考虑HSV颜色空间,常常导致颜色暗淡、饱和度对比不当以及出现伪影等结果。为解决这一问题,我们提出了一种新颖的草图上色方法——双颜色空间引导生成对抗网络(DCSGAN),该方法考虑了RGB和HSV颜色空间中包含的互补信息。具体而言,我们引入HSV颜色空间,通过一个学习从RGB到HSV颜色空间转换的颜色空间转换(CST)网络构建双颜色空间来监督我们的方法。然后,我们提出一种DP损失,使DCSGAN能够在像素级监督下生成色彩鲜艳的图像。此外,还设计了一种新颖的双颜色空间对抗(DCSA)损失,以在全局层面引导生成器减少伪影,满足观众的审美期望。大量实验和消融研究证明了所提方法优于先前的最先进(SOTA)方法。