在自己的数据集上实现CycleGAN(pytorch linux)_Iofal的博客-程序员秘密

代码
https://github.com/junyanz/pytorch-CycleGAN-and-pix2pix
参考文章
https://blog.csdn.net/Gavinmiaoc/article/details/80585531

1.将数据裁成256*256。

2.在datasets目录下新建文件夹label2image,在label2image下新建 trainA 和,trainB,testA,testB,A是你要训练的,B是你想生成的。(把图片放进去,不用成对)

3.在你的pytorch下,
pip install visdom
pip install dominate
python3 -m visdom.server
(and click the URL http://localhost:8097. To see more intermediate results, check out )
./checkpoints/maps_cyclegan/web/index.html

4.运行
python3 train.py --dataroot ./datasets/label2image --name label2image_cyclegan --model cycle_gan --pool_size 50 --no_dropout --gpu_ids 0
模型在checkpoint里

报错点一:
注意 这里–gpu_ids 0 是只有一个显卡,如果有两个上行改–gpu_ids 0,1
https://blog.csdn.net/weixin_43169773/article/details/109462068

报错点二:
ConnectionRefusedError: [Errno 111] Connection refused

https://blog.csdn.net/seamanj/article/details/102467626

This problem is related to visdom. You can start the visdom visualizatoin server by running python -m visdom.server. You can disable the visdom visualization by adding --display_id 0
这是因为可视化的问题,我没想看过程,就直接不可视化了,运行

python3 train.py --dataroot ./datasets/label2image --name label2image_cyclegan --model cycle_gan --pool_size 50 --no_dropout --gpu_ids 0 --display_id 0

报错点三:
python 引用 pillow 报错
ImportError: cannot import name ‘PILLOW_VERSION’

这个问题在引用 pillow 这个包时出现,主要原因是安装的 pillow 时 7.0.0 版本.

而 PILLOW_VERSION 在新版本中没有了。因此切换成旧版本(6.0.0)的 pillow 就可以了。

运行一下语句

pip uninstall pillow
pip install pillow==6
https://www.cnblogs.com/Assist/p/12427278.html

注意,最好先卸载旧版本,再安6.0版本,不然有可能运行时还是7的版本。

5.训练完的测试
运行
python test.py --dataroot ./datasets/label2image --name label2image_cyclegan --model cycle_gan --phase test --no_dropout
然后结果就在results里了

特别简单!!

版权声明:本文为博主原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明。
本文链接:https://blog.csdn.net/weixin_49636863/article/details/114499153

智能推荐

OpenCV—python 图像矫正(基于傅里叶变换—基于透视变换)_python-图像位置矫正_SongpingWang的博客-程序员秘密

一、基于傅里叶变换的图像校正由于拍照时,图像被旋转,为方便观看或使用,需要对旋转图像进行校正。旋转图像矫正流程思路借鉴 http://johnhany.net/2013/11/dft-based-text-rotation-correction/获取图像的傅里叶变换图二值化Hough直线检测计算倾斜角度旋转校正...

U8CO编写的API接口示例C#版本2.0后台-程序员秘密

新建项目web 项目新建控制器PuArrivalVouchController Info_PU.ClsS_Infor clsInfo = new Info_PU.ClsS_Infor(); VoucherVerify.UseMode usMode = VoucherVerify.UseMode.CS; VoucherCO_PU.clsVoucherCO_PUClass coPu = new VoucherCO_PU.clsVoucherCO_PUCla...

R用户的福音︱TensorFlow:TensorFlow的R接口_悟乙己的博客-程序员秘密

TensorFlowTensorFlow is an open source software library for numerical computation using data flow graphs. Nodes in the graph represent mathematical operations, while the graph edges represe

mac安装anaconda配置环境(pip安装包加速)_进击的挖掘机的博客-程序员秘密

一、苹果电脑安装anaconda1、官网安装anaconda安装完成后在Mac终端输入conda --version如果没有输出condo版本号,需要编辑路径。2、打开Mac终端,输入vim ~/.bash_profile 打开bash_profile文件点击i,进入insert模式,输入 export PATH=/anaconda3/bin:$PATH 点击Esc, 输入 :wq! ,保存退出编辑路径模式3、关闭终端,再次打开终端,输入conda --version,

java list控件_java.awt.List类:列表控件_冼辉的博客-程序员秘密

列表是指一次可以显示多个选项,同时一次也可以选择多个选项。在 Java AWT 中可以使用 List 类建立列表,建好后可以使用 add( ) 方法将选项置入列表。建立列表List 类继承 Component 类,下列是 List 类的构造方法。构造方法说明List( )默认显示 4 行的列表List(int rows)显示 rows 行的列表List(int rows,boolean multi...

callBack方法介绍_BlueElsL的博客-程序员秘密

ViewDragHelper中的Callback中常用方法介绍和创建

随便推点

gorm 的使用上的坑_gorm 坑_silver9886的博客-程序员秘密

func (a *RuleTarget) DelSystemRuleTargets( ctx context.Context, targets []*pbWeb.RuleTarget,) (successCount int32, err error) { db := database.GetDB(ctx) for _, target := range targets { // 注意如果是uid层面的删除,要级联删除cid和aid的数据 if...

深度学习多框架多平台推理引擎工具_框架工具引擎_踟蹰横渡口,彳亍上滩舟。的博客-程序员秘密

支持的计算平台:```bash- Windows 10 (Visual Studio 2019 x64)- Linux (x64, armv7, aarch64)- Android (armeabi-v7a, arm64-v8a)```支持的模型框架:```bash- TensorFlow Lite- TensorFlow Lite with delegate (XNNPACK, GPU, EdgeTPU, NNAPI)- TensorRT (GPU, DLA)- OpenCV(dn

安装成功后python报错_图文详解python之matplotlib安装与报错解决办法_袁冰涛的博客-程序员秘密

说明:首先python与matplotlib就不多介绍了,可以百度。个人也是第一次接触的小白,在安装过程中遇到一些较为难以理解的问题,通过努力解决后,在此做为例子分享出来。工具包下载官网地址:安装环境:windows 7 旗舰版 64位python 3.8.1具体步骤如下:一、先说一下简单的安装,也是最方便的,如果这种方法不行再试用第二步的方法。1、在cmd窗口输入如下命令:python -m ...

OSChina 周五乱弹 —— 夏日限定梦想_chengzheng3539的博客-程序员秘密

Osc乱弹歌单(2019)请戳(这里) 【今日歌曲】 小小编辑推荐:《And The Winner Is》- Gérard Darmon 《And The Winner Is》- Gérard Darmon 手机党少年们想听歌,请使劲儿戳(这里) @Pete_Jones :娃...

【已解决】Ubuntu18.04用python报错bash: ’/usr/bin/python2.7’: 没有那个文件或目录_吉柿wa的博客-程序员秘密

【已解决】Ubuntu18.04用python报错bash: ’/usr/bin/python2.7’: 没有那个文件或目录问题描述系统配置遇到的问题解决方法问题描述系统配置系统配置:Ubuntu18.04,Vmware14。遇到的问题我在系统上同时安装了python2.7与python3.6。当我使用 python指令与利用 python进行操作时出现如下错误:virtual-machine:~$ python//报错bash: ’/usr/bin/python2.7’: 没有那个文件

Tomcat启动失败错误解决Could not publish server configuration for Tomcat v7.0 Server at localhost_冰柠檬的小屋的博客-程序员秘密

今天又碰到了Tomcat启动失败的错误,错误提示如下: CouldnotpublishserverconfigurationforTomcatv7.0Serveratlocalhost. MultipleContextshaveapathof"/organ".意思很很清楚,有重名的节点,然后我根据百度的方法去tomcat安装目录conf下面的s...