在自己的数据集上实现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

智能推荐

android.view.WindowLeaked的解决方案_kentbleau的博客-程序员秘密

无意中发现windowleaked,错误详细信息如下:11-28 16:39:10.867: E/WindowManager(14064): android.view.WindowLeaked: Activity com.sbdz.agriculture.activity.ProductActivity has leaked window android.widget.TextView{4

百度地图Key的设置方法_user_key_recycled_水经注GIS的博客-程序员秘密

一、为什么要设置百度Key万能地图下载器提供了百度POI的下载功能,但由于本软件用户群极大,会导致一天之内访问量超出300万次以上而无法继续下载。因此,当POI下载不成功能,用户可以自己申请百度地图Key并在软件中设置之后,即可继续下载。点击以下链接,可以查看百度API访问的次数限制说明。http://lbsy

GBDT与XGboost小结_星青星晴的博客-程序员秘密

划重点——AdaBoost + 决策树 = 提升树                 Gradient Boosting + 决策树 = GBDTGBDT(Gradient Boosting Decision Tree)1.算法原理:GBDT(梯度提升决策树),是一种基于boosting串行式的集成学习方法,通过采用加法模型(即基函数的线性组合),以及不断减小训练过程产生的残差来达到将数据分类或者回...

python mysql实例_Python如何实现MySQL实例初始化详解_weixin_39633165的博客-程序员秘密

前言相信每位程序员对mysql应该都不陌生,MySQL是一个关系型数据库管理系统,由瑞典MySQL AB 公司开发,目前属于 Oracle 旗下产品。我们在日常开发中少不了要接触mysql。腾讯云上的mysql实例,初始化的时候,需要自己点击“初始化按钮”,然后还要手动输入16位复杂的密码,这个操作比较繁琐,而且如果你一次建立多个实例,这个操作量就很大了,所以必须想办法提高效率,于是就有了下文,批...

ansible下载配置及常见模块使用 _weixin_34311757的博客-程序员秘密

1.安装配置:yum install ansible主配置文件:/etc/ansible/ansible.cfg主机清单:/etc/ansible/hosts三个主程序:ansible,absible-doc,ansible-playbook2.定义主机清单:vim /etc/ansible/hosts可按组来分类3.使用ssh秘钥分发至定义好的主机:ss...

技术小白成长之路 - 谷歌云端 GCP Cloud Engineering - 第一篇 - 核心架构 Core Infrastructure_ChenYu here的博客-程序员秘密

谷歌云端 GCP Cloud Engineering - 第一篇 - 平台交互谷歌云端平台GCP简介二级目录三级目录1. 谷歌云端平台GCP资源层次结构2. VPC Network3. 存储与数据库Cloud Storage和Cloud BigtableCloud SQL和Cloud Spanner容器,K8S, 与K8S Engine4. --- App Engine云开发谷歌云端平台GCP简...

随便推点

华为光模块_weixin_34194359的博客-程序员秘密

华为技术有限公司是电信网络解决方案供应商。华为的主要营业范围是交换,传输,无线和数据通信类电信产品,在电信领域为世界各地的客户提供网络设备、服务和解决方案。 本文资料将具体介绍: 华为GBIC,SFP,XFP,XENPAK,X2光纤模块产品参数,产品型号。比如以下型号:SFP-GE-T SFP-GE-SX-MM850-A SFP-GE-LX-SM1310...

SSIS教程:创建简单的ETL包 -- 4. 增加错误处理流程(Adding Error Flow Redirection)_weixin_30265171的博客-程序员秘密

为了处理在转换过程中可能发生的错误,MicrosoftIntegrationServices允许根据每个组件和每个列来决定如何处理无法转换的数据。可以选择忽略某些列中的失败、重定向整个失败的行或者只是使组件失败。默认情况下,IntegrationServices中的所有组件被配置为在发生错误时失败。而使组件失败又会导致包失败,并使所有后续处理停止。如果不让失败导致包停止...

Ubuntu开机出现A start job is running for ....(1min,30s)因为gparted扩展分区导致swap的uuid发上改变_hzlarm的博客-程序员秘密

swap分区丢失的问题在将Ubuntu15.10升级Ubuntu16.04时,因系统提示磁盘容量不足时,gparted对其扩展分区遗留问题导致导致。扩展分区后swap的uuid发生改变,使用 sudo blkid 命令查看:然后sudo vi /etc/fstab把原来swap的uuid替换掉即可当时操作如下:步骤一:关机状态下在虚拟机设置中点击扩展步骤二使用gparted进行重...

servlet请求参数和返回页面内容的处理(HttpServletRequestWrapper与HttpServletResponseWrapper)..._小浣熊6616的博客-程序员秘密

        Servlet规范中的filter引入了一个功能强大的拦截模式。Filter能在request到达servlet的服务方法之前拦截HttpServletRequest对象,而在服务方法转移控制后又能拦截HttpServletResponse对象。        你可以使用filter来实现特定的任务,比如验证用户输入、请求参数以及压缩web内容等操作。还可以在response...

antDv 级联选择器(a-cascader)通过接口动态加载数据_antd a-cascader动态数据_踏浪而行生活圈的博客-程序员秘密

每次子选择器的内容都是通过将父级id传给接口获取到 最终保存到data中参数selectItem格式为 xxx,xx,x<a-cascader :options="options" :load-data="loadOptionsData" placeholder="请选择库位信息" change-on-select v-model="queryParam.positionCode" @change="onChange"/>...

提高 Vim 使用效率的 12 个技巧_vim 效率真的会高吗_guyue35的博客-程序员秘密

提高 Vim 使用效率的 12 个技巧 - 文章 - 伯乐在线  http://blog.jobbole.com/87481/首页最新文章IT 职场前端后端移动端数据库运维其他技术- 导航条 -首页最新文章IT 职场前端- JavaScript- HTML5- CSS后端- Python