UGAN代码使用

本文介绍了如何下载UGAN代码,进行数据集训练,设置超参数,保存模型checkpoint,并在训练后测试模型的过程。使用Tensorflow1.14环境,详细步骤包括配置训练和评估脚本,以及所需依赖。
摘要由CSDN通过智能技术生成

**# UGAN代码下载
链接:https://pan.baidu.com/s/1KTcXXnkUwsxlipWFmh5jSA
提取码:1234
复制链接下载UGAN代码
在这里插入图片描述

训练你自己的数据集

解压代码找到train.py
train.py
设置超参数
设置你的超参数
存放cheeckpoint和结果保存的路径
存放cheeckpoint和结果保存的路径,如果没有对应的文件夹,新建即可,也可以修改保存路径,为了简便这里不做改动。
这里没带引号的是你之前设置的超参数的值。

在这里插入图片描述
保存pickle文件,不熟悉的朋友可以不用管他
模型定义
定义生成器和判别器
定义loss定义loss
日志和优化器在这里插入图片描述日志和优化器
载入预训练模型
训练数据路径
在两个glob之后加上当前任务数据集存放位置,代码中为默认读取地址
开始训练
从这里开始训练模型,总共迭代EPOCHS次
在这里插入图片描述
保存模型checkpoint
了解完模型训练过程,设置好超参数等,就可以开始训练了。运行train.py即可

测试模型

训练完自己的模型,找到eval.py文件
在这里插入图片描述
在这里插入图片描述

跟训练一样的过程,这里从训练时保存的pickle文件中读取超参数。当然也可以自己按照train.py里那样自己定义,为了简便直接读取pickle文件了。
在这里插入图片描述
这里设置checkpoint读取路径和结果保存路径,值都是从pickle文件中读取的如果是自己训练的可以忽略。IMAGES_DIR是结果保存路劲,可以在EXPERIMENT_DIR文件目录下新建一个文件夹用于存放结果。
在这里插入图片描述
定义模型,读取checkpoint
在这里插入图片描述

设置测试文件夹路径,支持从文件形式包括在蓝圈中,在橙色圈中设置你的测试数据路径
在这里插入图片描述
开始测试
在这里插入图片描述
使用misc.imsave保存测试结果,计算模型速度
代码解释过程结束,运行程序即可。

模型使用的是Tensorflow1.14,1.1x环境应该都能运行,misc需要较低版本的scipy,高版本中的scipy中已删除这个包。

  • 7
    点赞
  • 13
    收藏
    觉得还不错? 一键收藏
  • 15
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 15
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值