在自己的数据集上实现CycleGAN(pytorch linux)

代码
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里了

特别简单!!

你可能感兴趣的:(在自己的数据集上实现CycleGAN(pytorch linux))