基于新的风格损失函数与亮度信息的图像风格迁移算法

摘要

针对传统图像风格化处理算法在进行图像风格化处理过程中,出现颜色信息丢失、笔触细节错乱以及边缘线条扭曲等情况,本文基于卷积神经网络的图像风格迁移算法进行了改进,通过综合考虑图像的亮度颜色信息特性,提出一种基于新的风格损失函数与亮度信息的图像风格迁移算法。首先,将内容图像进行色彩空间转换,让风格转移单独在亮度通道上进行,保留了内容图像的色彩信息。其次,把直方图损失加入到总的损失函数中,构造新的总损失函数,提高了风格化图像的稳定性。最后,把提取的内容图像边缘信息叠加到结果图中,达到改善风格化结果图边缘信息模糊不清的目的。实验表明,改进后的方法在笔触细节、纹理结构以及色彩空间上有更好的整体效果表现。