用Python2跑Cycle-GAN模型（Pytorch）

最新推荐文章于 2023-02-14 12:02:38 发布

ccc_irene

最新推荐文章于 2023-02-14 12:02:38 发布

阅读量965

点赞数 1

分类专栏：机器学习文章标签：机器学习

本文链接：https://blog.csdn.net/qq_41433316/article/details/102713177

版权

别问我为什么要用python2跑，毕竟官方都要放弃了。。。
步骤网上一堆，此次跑模型也算是先练习一下，积累一点实战经验
代码：GitHub-pytorch
本次选用的数据集是经典的horse2zebra数据集
如果想按照教程给的方式下载，可以试试下面的命令：

bash ./scripts/download_cyclegan_model.sh horse2zebra

如果你运气好或者熟练掌握翻墙技术，就可以，如果实在是下不下来，这里我给出我的百度云链接：c4qh
如果它失效了评论区说一声
弄完后就可以执行下面的命令了：

python train.py --dataroot ./datasets/maps --name maps_cyclegan --model cycle_gan

下面分享一下我在训练中遇到的问题：
1、ImportError：cannot import name ABC
在这里插入图片描述
参考 https://blog.csdn.net/weixin_45250844/article/details/94966415
毕竟是python2，很多地方都不一样，改改代码就好
这个地方还挺多，主要是下面两个文件里：
（注意一下加python版本的地方）
base_model.py

# from abc import ABC, abstractmethod   python3
from abc import ABCMeta, abstractmethod         #python2.7
from . import networks


# class BaseModel(ABC):         python3