基于带跳跃连接与融合层的 U-Net 的风景图像彩色化
计算机视觉与模式识别
2022-05-26 v1 机器学习
图像与视频处理
摘要
我们提出了一种结合 U-Net 模型与融合层特征来自动为灰度图像上色的新型技术。该方法使模型能够从预训练的 U-Net 中学习图像彩色化。此外,融合层用于将依赖于小图像块的局部信息结果与每一类上整幅图像的全局先验相合并,形成视觉上更具说服力的彩色化结果。最后,我们通过用户研究评估验证了我们的方法,并与最先进(SOTA)方法进行比较,取得了改进。
引用
@article{arxiv.2205.12867,
title = {Image Colorization using U-Net with Skip Connections and Fusion Layer on Landscape Images},
author = {Muhammad Hisyam Zayd and Novanto Yudistira and Randy Cahya Wihandika},
journal= {arXiv preprint arXiv:2205.12867},
year = {2022}
}