码农知识堂 - 1000bd
  •   Python
  •   PHP
  •   JS/TS
  •   JAVA
  •   C/C++
  •   C#
  •   GO
  •   Kotlin
  •   Swift
  • 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

  • 相关阅读:
    【图论——第十讲】匈牙利算法实现二分图的最大匹配
    基于流的深度生成模型
    LVS负载群集--DR模式
    6. 毕业设计温湿度监控系统(ESP8266 + DHT11 +OLED 实时上传温湿度数据给公网服务器并在OLED显示屏上显示实时温湿度)
    Unity Shader - if 和 keyword 的指令比较
    可制造性评估(DFM)
    基于注解管理bean(功能分析、注解和扫描、扫描组件、bean的id)
    Task Scheduler
    MySQL学习记录 —— 십구 命令执行
    10月第4周榜单丨飞瓜数据B站UP主排行榜(哔哩哔哩平台)发布!
  • 原文地址:https://blog.csdn.net/m0_47709941/article/details/127736332
  • 最新文章
  • 沪漂五周年了:我越来越迷茫了
    Agentic Skill Routing 实战:别再把所有 Skill 塞进 AI Agent 上下文
    MySQL-Seconds_behind_master的精度误差
    [MAF预定义ChatClient中间件-03]CachingChatClient——利用缓存省钱省时间
    AI的至暗历史:从万众期待到被政府撤资,AI的两次死亡徘徊
    Agent OS :五种驯服不确定性的范式
    PortSwigger SQL注入LAB11
    数据库即时编译JIT
    [Begin]AI Learn Data Day 0
    深度学习进阶(二十七)现代 LLM 的核心架构设计其二:SwiGLU
  • 热门文章
  • 十款代码表白小特效 一个比一个浪漫 赶紧收藏起来吧!!!
    奉劝各位学弟学妹们,该打造你的技术影响力了!
    五年了,我在 CSDN 的两个一百万。
    Java俄罗斯方块,老程序员花了一个周末,连接中学年代!
    面试官都震惊,你这网络基础可以啊!
    你真的会用百度吗?我不信 — 那些不为人知的搜索引擎语法
    心情不好的时候,用 Python 画棵樱花树送给自己吧
    通宵一晚做出来的一款类似CS的第一人称射击游戏Demo!原来做游戏也不是很难,连憨憨学妹都学会了!
    13 万字 C 语言从入门到精通保姆级教程2021 年版
    10行代码集2000张美女图,Python爬虫120例,再上征途
小工具 小游戏
Copyright © 2022 侵权请联系2656653265@qq.com    京ICP备2022015340号-1

京公网安备 11010502049817号