计算机工程

作者：陈志鹏, 郑文秀, 黄琼丹

单位：西安邮电大学 通信与信息工程学院, 西安 710121

关键词：图像风格转换,边缘检测,Sobel滤波器,卷积神经网络,均方误差

基金：陕西省重点研发计划项目（2018GY-150）。

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

来源：2021年第12期

《计算机工程》期刊编辑部

查看计算机工程杂志2021年第12期