文件名称:ImgFastNeuralStyleTransfer_TensorFlow:快速风格迁移学习实践
文件大小:1.34MB
文件格式:ZIP
更新时间:2024-06-07 16:08:04
tensorflow style-transfer Python
快速风格迁移实践 可用资源下载 训练模型为slim预训练资源 本次实验仅仅是用来训练数据(不含标签),后面的5个链接是为了数据完整性给出的,针对本次实验不需下载 项目文件介绍 net:放置官方使用slim模块实现的网络模型 train2017:下面存放图片数据,即COCO训练用数据(无标签) general_net.py:用于生成风格迁移图片的网络 losses.py:损失函数集合,包含风格损失计算,内容损失计算,以及平滑损失计算 main.py:主文件,整合了整个网络图景,并含有训练过程的逻辑 style_image_gen.py:使用训练好的模型,将自己的图片加载上进行风格迁移 utils.py:辅助函数,包含图片预处理、gram矩阵计算以及风格图片gram矩阵获得等函数 脚本运行指南 训练和迁移自己的图片都很简单,训练时如下即可: python main.py 然后新建终端,输入,
【文件预览】:
ImgFastNeuralStyleTransfer_TensorFlow-master
----网络结构示意图.png(548KB)
----img()
--------mosaic.jpg(76KB)
----utils.py(6KB)
----train2017()
--------000000000030.jpg(70KB)
----style_image_gen.py(1KB)
----main.py(10KB)
----nets()
--------vgg.py(10KB)
----losses.py(2KB)
----general_net.py(7KB)
----README.md(4KB)
----pretrained()
--------此处存放预训练模型(0B)
----效果示意图.png(759KB)