实战背景

基于公司做AI设计理论测试,在前期对GAN技术的了解基础上,实战跑了模型生成AI图案花型设计的效果,沉淀AI智能解决方案。测试电脑性能不好,连续跑了25天,实战不能再等了,就跑了33200次循环,效果可以看出来了!

数据集整理

难点在于网络上没有专门的图案款式数据集,寻找了一些趋势网站,付费购买了花型数据;另外一个思路是直接去面料网站寻找(太花时间,没有我那个这个方向走,很多电商图片或花型网站都有水印,不好解决),考虑需要背景干净且为白色的图片,降低噪点,于是就有了数据集制作的过程。

数据集制作思路:看看代码 → 了解数据集图片大小 → 下载图片 → 归类图片 → 清洗图片 → 压缩大小

数据集规格,制作并提供了25316张,像素为600*600的白色背景图片,分辨率为380dpi,10个大类的图案花型

花型图案数据集.png

代码说明

直接使用之前动漫训练模型,原模型地址:
https://www.16084a.com/index.php/net/gandm.html

实战在次基础上有优化和调整参数

训练效果

200次循环
train_00000200.png

3000次循环
train_00003000.png

6000次循环
train_00006000.png

16000次循环
train_00016000.png

20000次循环
train_00020000.png

30000次循环
train_00030000.png

33200次循环
train_00033000.png

最后修改:2019 年 12 月 14 日
文章有用,请随意赞赏