快速风格迁移实践
可用资源下载
训练模型为slim预训练资源
本次实验仅仅是用来训练数据(不含标签),后面的5个链接是为了数据完整性给出的,针对本次实验不需下载
项目文件介绍
net:放置官方使用slim模块实现的网络模型
train2017:下面存放图片数据,即COCO训练用数据(无标签)
general_net.py:用于生成风格迁移图片的网络
losses.py:损失函数集合,包含风格损失计算,内容损失计算,以及平滑损失计算
main.py:主文件,整合了整个网络图景,并含有训练过程的逻辑
style_image_gen.py:使用训练好的模型,将自己的图片加载上进行风格迁移
utils.py:辅助函数,包含图片预处理、gram矩阵计算以及风格图片gram矩阵获得等函数
脚本运行指南
训练和迁移自己的图片都很简单,训练时如下即可:
python main.py
然后新建终端,输入,
评论0
最新资源