1.项目基于 MNIST 数据集,使用 VGG-19 网络模型,将图像进行风格迁移,实现去噪功能。
2.项目运行环境:Python 和 TensorFlow 运行环境。需要 Python 3.6 及以上配置,使用conda安装环境 conda create -n tensorflow python=3.8.10
3.项目包括 3 个模块:图片处理、模型构造、迭代更新。项目用到的网络模型为预训练好的VGG-19,使用过程中抛弃最后三个全连接层,取出前面各层的参数,构建网络结构。损失函数,由内容损失、风格损失构成。内容损失采用 L2范数损失,风格损失用 Gram 矩阵计算各通道的相关性,以便更好的捕捉笔触、纹理等细节信息,利用 adam 梯度下降算法进行优化。
4.准确率评估:对于图像风格迁移这种模糊算法,并没有客观的评判标准。损失函数可以反映出一部分情况,更多的是人为观察运行结果。经测试,经过 40 次迭代风格迁移已很明显,可根据自身需求,合理调节迭代次数。
1