基于Sobel滤波器的图像风格转换算法

摘要

基于迭代的图像风格转换在图像重组时未考虑内容图像的结构,导致生成的图像存在线条扭曲.为约束图像重组时的信息,提出一种基于边缘检测的图像风格转换算法.通过Sobel滤波器在内容图像和生成图像相同的卷积层上提取边缘信息,同时以均方误差作为损失函数.在此基础上,将边缘损失、内容损失和风格损失的加权代数和作为神经网络的总损失.实验结果表明,该算法能够有效抑制图像的线条扭曲,减少图像噪声,生成更高质量的图像.