pytorch-CycleGAN-and-pix2pix-master的操作使用

我针对如何使用pytorch-CycleGAN-and-pix2pix-master的代码进行了简单讲解,代码链接在下:

https://github.com/junyanz/pytorch-CycleGAN-and-pix2pix

visdom可视化

首先启动visdom可视化,可以看到loss变化等

python -m visdom.server

然后在本地网页打开

http://localhost:8097

即可看到随时变化数据可视化

这里我们要注意,如果我们不打开visdom的话,可能程序会中途意外停止,所以建议打开visdom,无论是否要看loss变化。

数据处理

通过下载官方数据集,我们发现cyclegan和pix2pix数据集是不一样的;

cyclegan的很简单,就是trainA,trainB,testA,testB。

而pix2pix则需要两张size相同的对应图片左右拼接对齐,这里我写了一个pix2pix数据集制作代码,很简单即可生成。

pix2pix需要train,test,两个文件夹数据集。

cycleGAN

  • 训练

在terminal中输入

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

其中,第一个--后面的是数据集位置,第二个--的是训练出来的模型,存放的文件夹名称(在checkpoints文件夹中,后面会说到),第三个--是使用的模型。

这样模型就训练好了,在模型的训练期间或者训练结束后,在checkpoints文件夹中打开web文件夹,可以看到实时训练的结果,同时checkpoints文件夹中的这些pth文件,就是我们所训练出来的模型。

  • 测试

在terminal中输入

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

其中,第一个--后面的是数据集位置,第二个--的是训练出来的模型,存放的文件夹名称(和上面一样,就是checkpoints文件夹中的那个),第三个--是使用的模型。

这里我们测试完成后,就可以在result文件夹中看到结果。

这里我们会发现有很多的结果,可是我们就想要我们生成器所训练的最终结果,那我们该怎么办呢?

这里我们需要在checkpoints中创建一个全新的文件夹,文件夹名称可以随意设置,我的建议是在之前训练的时候的那个文件夹名称后面加上_predect,比如maps_cyclegan_predect,这样我们创建好文件夹之后,我们从maps_cyclegan中找到latest_net_G_A.pth,这是last训练出来的生成器模型,将其复制到我们创建好的文件夹maps_cyclegan_predect中,并且将latest_net_G_A.pth改名为latest_net_G.pth

然后我们再在terminal中输入:

python test.py --dataroot datasets/maps --name maps_cyclegan_pretrained --model test --no_dropout

这样我们就可以在resule中的maps_cyclegan_pretrained文件中看到只有训练完成的生成器所生成的结果啦。

pix2pix

同上一样,我们通过同样的方式打开visdom,查看数据可视化。

  • 训练

在terminal中输入

python train.py --dataroot ./datasets/facades --name facades_pix2pix --model pix2pix --direction BtoA

  • 测试

再在terminal中输入

python test.py --dataroot ./datasets/facades --name facades_pix2pix --model pix2pix --direction BtoA

测试完成后就可以在result中看到结果了

同样的方式,我们也可以单独提取出来latest_net_G.pth,放在一个单独的_predect文件夹中

然后我们再在terminal中输入:

python train.py --dataroot ./datasets/facades --name facades_pix2pix_predect --model pix2pix --direction BtoA

就可以在result中看到结果了。

posted @ 2023-05-16 16:03  top王  阅读(686)  评论(0编辑  收藏  举报