Color2Embed:基于颜色嵌入的快速示例图像着色
计算机视觉与模式识别
2021-08-25 v3 多媒体
摘要
本文提出一种利用颜色嵌入的快速示例图像着色方法,称为 Color2Embed。通常,由于难以获取输入与真值图像对,以无监督、非配对训练方式训练示例着色模型十分困难。现有算法通常致力于两个步骤:i) 检索大量高相似度参考图像以准备训练数据集,这不可避免地耗时且繁琐;ii) 设计复杂模块,通过计算并利用参考图像与目标图像之间的深层语义对应(如非局部操作)将参考图像的颜色迁移至目标图像,这在测试时计算代价高昂。与以往方法不同,我们采用自增强的自参考学习方案,其中参考图像由原始彩色图像经图形变换生成,从而可将训练构建为配对形式。其次,为减少处理时间,我们的方法显式提取颜色嵌入并利用渐进式风格特征变换网络,将颜色嵌入注入最终图像的重建中。该设计更轻量且可解释,以快速处理速度取得了令人满意的性能。
引用
@article{arxiv.2106.08017,
title = {Color2Embed: Fast Exemplar-Based Image Colorization using Color Embeddings},
author = {Hengyuan Zhao and Wenhao Wu and Yihao Liu and Dongliang He},
journal= {arXiv preprint arXiv:2106.08017},
year = {2021}
}
备注
10 pages, 10 figures