gan与dcgan训练自己的数据集
创始人
2024-02-27 14:21:11
0

gan
https://blog.csdn.net/weixin_50113231/article/details/122959899
dcgan
源码地址:https://github.com/carpedm20/DCGAN-tensorflow

安装教程

环境配置
将代码克隆到本地后首先按照官网所需依赖环境进行配置
在这里插入图片描述
由于该文章比较早所以python与tensorflow最好按照要求进行安装
注意:
tensorflow0.12.1要装对应系统cpu版本,moviepy也要安装低版本的,这里给出我的成功运行配置的环境依赖
在这里插入图片描述

数据集准备

一般都是想训练自己的数据集,现成的能运行也没啥意义,这里就说下训练自己的数据集吧
最简单的就是再源代码根目录创建data文件夹,里面包含不同文件名的图片文件夹,例如:
在这里插入图片描述
只需要图片就可以了,不需要标签

训练过程

训练自己的数据集需要更改一些参数,按照官网的就行
在这里插入图片描述
一般需要设置epoch ,batch_size,out_dir,dataset,input_fname_pattern等参数,另外训练的时候加上–train,需要裁剪图片的话加上–crop与input_height,output_height,也要加上visualize进行可视化,可以查看不同批次的训练结果。

训练遇到的问题以及解决办法

1.TypeError: ‘_FlagValues’ object is not iterable
问题是main.py文件的71行,经过打印确实FLAGS = flags.FLAGS不是一个字典,只是一个对象,估计版本问题,网上没搜到答案,既然如此,明白代码意思就是将一些字典参数写入到json文件中
在这里插入图片描述
文件开头在控制台实际已经打印出字典信息,直接拿来替换即可
在这里插入图片描述
2.“one pic error!..”
在训练的时候进行可视化,却没有保存图片,按照这个博客找到答案
在这里插入图片描述
出错地方为model.py文件中image_manifold_size这个函数限制条件,在utils文件中最后一段
在这里插入图片描述

其实就是batch_size设置的问题,因为代码中有开放操作,所以batch_size设置最好是能被开放整数,例如4,16,32等,另外也要注意图片总的个数大于且是batch_size的整数倍。
3.TypeError: Cannot handle this data type
按照这个修改文件代码
在这里插入图片描述
重点:
np.uint8一定要加上,其实transform部分的if crop也可以不修改,最后直接将return部分改为return cropped_image,一个意思,这是针对裁剪来说,如果不裁剪就需要按照下面代码执行
在这里插入图片描述
4.SyntaxError: Non-ASCII character ‘\xe6’
代码中有中文字符需要在代码开头加上#encoding: utf-8

相关内容

热门资讯

喜欢穿一身黑的男生性格(喜欢穿... 今天百科达人给各位分享喜欢穿一身黑的男生性格的知识,其中也会对喜欢穿一身黑衣服的男人人好相处吗进行解...
发春是什么意思(思春和发春是什... 本篇文章极速百科给大家谈谈发春是什么意思,以及思春和发春是什么意思对应的知识点,希望对各位有所帮助,...
网络用语zl是什么意思(zl是... 今天给各位分享网络用语zl是什么意思的知识,其中也会对zl是啥意思是什么网络用语进行解释,如果能碰巧...
为什么酷狗音乐自己唱的歌不能下... 本篇文章极速百科小编给大家谈谈为什么酷狗音乐自己唱的歌不能下载到本地?,以及为什么酷狗下载的歌曲不是...
家里可以做假山养金鱼吗(假山能... 今天百科达人给各位分享家里可以做假山养金鱼吗的知识,其中也会对假山能放鱼缸里吗进行解释,如果能碰巧解...
华为下载未安装的文件去哪找(华... 今天百科达人给各位分享华为下载未安装的文件去哪找的知识,其中也会对华为下载未安装的文件去哪找到进行解...
四分五裂是什么生肖什么动物(四... 本篇文章极速百科小编给大家谈谈四分五裂是什么生肖什么动物,以及四分五裂打一生肖是什么对应的知识点,希...
怎么往应用助手里添加应用(应用... 今天百科达人给各位分享怎么往应用助手里添加应用的知识,其中也会对应用助手怎么添加微信进行解释,如果能...
苏州离哪个飞机场近(苏州离哪个... 本篇文章极速百科小编给大家谈谈苏州离哪个飞机场近,以及苏州离哪个飞机场近点对应的知识点,希望对各位有...
客厅放八骏马摆件可以吗(家里摆... 今天给各位分享客厅放八骏马摆件可以吗的知识,其中也会对家里摆八骏马摆件好吗进行解释,如果能碰巧解决你...