【深度学习】CycleGAN开源项目学习笔记 | 完整流程 | 报错总结 | pytorch
文章目录
- 前言
- 一、下载项目，文件结构观察
- 二、数据集下载
- 三、训练
-
- 3.1、训练初体验
- 3.2、命令行输入参数
- 3.3、继续训练命令
- 四、预测
-
- 4.1、使用自己训练完的权重文件进行预测
- 4.2、使用网上的预训练文件进行预测
前言
你敢想象，就是这么一个简单的开源网站，我居然调了一天才通（毕竟是第一次，一些比如visdom，命令行的参数第一次接触到）
本文所使用的开源项目网址：https://github.com/junyanz/pytorch-CycleGAN-and-pix2pix
一、下载项目，文件结构观察
下载项目完成后，我们看看内部的结构：
据说后面很多开源项目都是按照这个格式来的（大致格式）
二、数据集下载
在ReadMe里面已经提到了数据集下载方式：
但是这里的bash命令一直没有运行成功，所以换用其他方法：
我们打开dataset文件夹：
用记事本打开里面的cyclegan的sh文件，把这行网址复制粘贴（最后一个反斜杠后面的内容删除）：
打开网址，选择你想要的数据集，这里我们的实验选择斑马horse2zebra数据集:
解压之后，把数据集粘贴到datasets下面：
斑马数据集内部长这个样子，后面如果要自己制造数据集也是这样搞：
三、训练
3.1、训练初体验
训练前，看看自己的环境是不是安装好了，尤其是visdom这个可视化库。没用过的话建议先用用看，试一试，防止后面出问题找不出来错误。
在开始训练前，需要打开visdom，在命令行直接输入以下命令：
python -m visdom.server
点击下面的8097网址进去：
下面就是比较关键的，输入训练指令，Readme里面也给出了比较清楚的指令（不过我们需要进行修改）：
python train.py --dataroot ./datasets/maps --name maps_cyclegan --model cycle_gan
我们来解释一下，这三个需要导入的参数分别是什么：？
1）datasets 这个最简单，就是数据集的路径
2）name 这里指的是保存训练权重的文件夹的名字。在开始训练后，文件夹里会生成一个checkpoints子文件夹，用来保存权重文件：
而打开后就是我们命名的文件夹：
里面的权重文件长这样：
web里面放着训练过程
3) model：这个顾名思义就是训练的模型是啥，这个开源项目的模型如下（在model文件夹里）：
在命令行里输入cycle_gan后 程序会自动加上_model
介绍完之后，我们来更改
python train.py --dataroot ./datasets/horse2zebra --name horse2zebra --model cycle_gan
我们在命令行里面运行，不出意外的话，意外就来了：
报错内容：module ‘torch._C’ has no attribute ‘_cuda_setDevice’
在网上查找资料说，这是因为环境里pytorch下载的CPU版本导致的。但是我环境里明明是GPU版本呀，这里有可能是因为CMD命令行识别环境错了，所以我们换在anaconda里面运行：
记得要激活自己的GPU环境，同时切换到文件路径，再输入命令后：
这里开始，训练就正式开始了，训练可视化可以在visdom里面看：
3.2、命令行输入参数
我们在3.1中输入到命令行的参数为：
python train.py --dataroot ./datasets/horse2zebra --name horse2zebra --model cycle_gan
也就三个。其实事实上能输入的不止这些，还有很多其他参数也是可以输入进去的：
在train_options这个py文件中，存在大量的add_argument，这些参数都是可以输入的，且有文字描述，帮助大家理解它们是用来干啥的。
3.3、继续训练命令
比如我昨天训练了15个epoch，今天我想继续训练，可以在命令行中输入下列指令：
从图片中我们可以看到，继续训练的开关被打开了
但是它画图依然是从1epoch开始画的，而不是从我停止训练的15epoch，不知道有没有大神能够指点一下
四、预测
4.1、使用自己训练完的权重文件进行预测
这里我自己训练了15轮，权重文件保存在checkpoint里：
我们在anconda命令行里输入：
python test.py --dataroot ./datasets/horse2zebra --name horse2zebra --model cycle_gan
结果会保存在results 文件夹里：
打开html我们看看效果：
15轮的效果还是一般 轮数没训练够
4.2、使用网上的预训练文件进行预测
首先我们需要在网上下载预训练权重文件：预训练文件
下载完毕后，在check_points文件夹里面新建一个文件夹：horse2zebra.pth_pretrained：
将刚刚下载的预训练权重重新命名为：latest_net_G.pth：
这时候的准备工作已经完成了，此时打开anconda 激活环境 切换路径，输入下列指令：
python test.py --dataroot datasets/horse2zebra/testA --name horse2zebra.pth_pretrained --model test --no_dropout
此时就开始预测了：
预测完成后可以在根目录下的results文件夹中找到结果：
载打开里面的文件夹可以看到
打开html:
可以看到效果明显比我自己训练的好（毕竟我才训练了15轮）