图像风格迁移、色彩填充与色彩变换等,严格意义上来说都属于计算机视觉任务中图像处理的分支。它们输入的是图像,输出的也是图像,过程实现图像到图像的内容与风格的转换,深度学习在这类图像处理任务上也取得了良好的效果。在DNN模块中支持常见风格迁移的图像转换模型,该模型是李飞飞等人发表的感知损失实时分割迁移与超分辨率论文的torch版本实现,模型的下载地址为:
https://github.com/jcjohnson/fast-neural-style
模型介绍
模型支持任意尺寸的图像输入,输出NCHW的四维数据,其中N=1,C=3表示彩色图像,H跟W分别表示图像的高与宽。作者提供了很多种预训练的风格迁移模型以供读者使用,这里下载了下面九种风格转换的预训练模型:
composition_vii.t7
starry_night.t7
la_muse.t7
the_wave.t7
mosaic.t7
the_scream.t7
feathers.t7
candy.t7
udnie.t7

这些模型都是torch框架支持的二进制权重文件,加载模型之后,就可以调用得到结果,通过对输出结果反向加上均值,到0~255的RGB色彩空间,即可得到转换后的风格图像,如图12-8所示,是九种变换风格的效果演示。

以上演示来自《应用开发:入门、进阶与工程化实践》一书第十二章 第五小节内容。
关于新书
《应用开发:入门、进阶与工程化实践》全书共计16个章节,重点聚焦开发常用模块详解与工程化开发实践,提升应用开发能力,助力读者成为开发者,同时包含深度学习模型训练与部署加速等知识,帮助开发者进一步拓展技能地图,满足工业项目落地所需技能提升。购买请点链接:
https://item.jd.com/10092255924058.html

学习课程有专属答疑群
负责贴身答疑解惑
读者专属QQ群 :
进群暗号:读者

© 版权声明
文章版权归作者所有,未经允许请勿转载。
相关文章
暂无评论...