您的位置:首页 > 理论基础 > 计算机网络

卷积神经网络图像纹理合成 Texture Synthesis Using Convolutional Neural Networks

2020-11-19 10:18 621 查看

代码实现

概述

这是关于Texture Synthesis Using Convolutional Neural Networks论文的tensorflow2.0代码实现,使用keras预训练的VGG19模型,依照论文重新更改了vgg19的设置。
本实现在论文给出的纹理中基本能得到复现结果

环境

tensorflow >2.0
numpy
Pillow
matplotlib

Usage

python3 synthesize.py src.jpg --output output.jpg

文件

  • custome_vgg.py
    对vgg19重做修改
  • utils.py
    文件读写,格拉姆矩阵计算等
  • Texture Synthesis Using Convolutional Neural Networks.ipynb 关于代码实现的实验代码和思路讲解

result




讨论

  • 对于合成纹理中的高频噪声可以加入总变分损失(Total variation loss)优化,但笔者由于调参原因找不到合适的参数,因此放弃
  • 同时,使用直方图匹配在深色图片上也会起到优化作用,但泛化性不强
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: 
相关文章推荐