基于Tensorflow.js的DIY图像风格迁移系统开发

Development of DIY image style migration system based on Tensorflow.js

针对目前各类美图软件美化效果单一、风格雷同的问题,文章开发一套基于快速风格迁移的DIY图像风格迁移系统。系统通过COCOS游戏引擎和JS等工具完成网页的UI设计与数据的获取处理,利用深度学习技术,使用Google公司的Tensorflow.js开源框架搭建风格迁移网络,并完成模型迁移及对JS获取到的风格图像、内容图像数据的转换,实现将用户选定图片的风格迁移到指定的图片上。同时,系统支持用户指定再训练轮次并对模型重新离线训练,调整风格迁移效果。

Aiming at the problems of single beautification effect and similar styles of various Meitu softwares,a set of DIY image style migration system based on rapid style migration was developed.Use COCOS game engine and JS tools to complete the UI design and data acquisition and processing of web pages,use deep learning technology,use Google’s Tensorflow.js open source framework to build a style migration network,and complete model migration and style images acquired by JS.The conversion of content image data realizes the transfer of the style of the picture selected by the user to the specified picture.At the same time,the system supports users to specify retraining rounds and retrain the model offline to adjust the style transfer effect.