gan与dcgan训练自己的数据集

2023-10-21 23:59
文章标签 数据 训练 gan dcgan

本文主要是介绍gan与dcgan训练自己的数据集,希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!

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

这篇关于gan与dcgan训练自己的数据集的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!



http://www.chinasem.cn/article/257793

相关文章

Linux下利用select实现串口数据读取过程

《Linux下利用select实现串口数据读取过程》文章介绍Linux中使用select、poll或epoll实现串口数据读取,通过I/O多路复用机制在数据到达时触发读取,避免持续轮询,示例代码展示设... 目录示例代码(使用select实现)代码解释总结在 linux 系统里,我们可以借助 select、

C#使用iText获取PDF的trailer数据的代码示例

《C#使用iText获取PDF的trailer数据的代码示例》开发程序debug的时候,看到了PDF有个trailer数据,挺有意思,于是考虑用代码把它读出来,那么就用到我们常用的iText框架了,所... 目录引言iText 核心概念C# 代码示例步骤 1: 确保已安装 iText步骤 2: C# 代码程

Pandas处理缺失数据的方式汇总

《Pandas处理缺失数据的方式汇总》许多教程中的数据与现实世界中的数据有很大不同,现实世界中的数据很少是干净且同质的,本文我们将讨论处理缺失数据的一些常规注意事项,了解Pandas如何表示缺失数据,... 目录缺失数据约定的权衡Pandas 中的缺失数据None 作为哨兵值NaN:缺失的数值数据Panda

C++中处理文本数据char与string的终极对比指南

《C++中处理文本数据char与string的终极对比指南》在C++编程中char和string是两种用于处理字符数据的类型,但它们在使用方式和功能上有显著的不同,:本文主要介绍C++中处理文本数... 目录1. 基本定义与本质2. 内存管理3. 操作与功能4. 性能特点5. 使用场景6. 相互转换核心区别

python库pydantic数据验证和设置管理库的用途

《python库pydantic数据验证和设置管理库的用途》pydantic是一个用于数据验证和设置管理的Python库,它主要利用Python类型注解来定义数据模型的结构和验证规则,本文给大家介绍p... 目录主要特点和用途:Field数值验证参数总结pydantic 是一个让你能够 confidentl

JAVA实现亿级千万级数据顺序导出的示例代码

《JAVA实现亿级千万级数据顺序导出的示例代码》本文主要介绍了JAVA实现亿级千万级数据顺序导出的示例代码,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面... 前提:主要考虑控制内存占用空间,避免出现同时导出,导致主程序OOM问题。实现思路:A.启用线程池

SpringBoot分段处理List集合多线程批量插入数据方式

《SpringBoot分段处理List集合多线程批量插入数据方式》文章介绍如何处理大数据量List批量插入数据库的优化方案:通过拆分List并分配独立线程处理,结合Spring线程池与异步方法提升效率... 目录项目场景解决方案1.实体类2.Mapper3.spring容器注入线程池bejsan对象4.创建

PHP轻松处理千万行数据的方法详解

《PHP轻松处理千万行数据的方法详解》说到处理大数据集,PHP通常不是第一个想到的语言,但如果你曾经需要处理数百万行数据而不让服务器崩溃或内存耗尽,你就会知道PHP用对了工具有多强大,下面小编就... 目录问题的本质php 中的数据流处理:为什么必不可少生成器:内存高效的迭代方式流量控制:避免系统过载一次性

C#实现千万数据秒级导入的代码

《C#实现千万数据秒级导入的代码》在实际开发中excel导入很常见,现代社会中很容易遇到大数据处理业务,所以本文我就给大家分享一下千万数据秒级导入怎么实现,文中有详细的代码示例供大家参考,需要的朋友可... 目录前言一、数据存储二、处理逻辑优化前代码处理逻辑优化后的代码总结前言在实际开发中excel导入很

MyBatis-plus处理存储json数据过程

《MyBatis-plus处理存储json数据过程》文章介绍MyBatis-Plus3.4.21处理对象与集合的差异:对象可用内置Handler配合autoResultMap,集合需自定义处理器继承F... 目录1、如果是对象2、如果需要转换的是List集合总结对象和集合分两种情况处理,目前我用的MP的版本