pix2pix踩坑

找了个 pytorch 实现的 pix2pix 踩了不少坑==

用的这个人的:https://github.com/junyanz/pytorch-CycleGAN-and-pix2pix

注意 doc 里面说可以怎么怎么样,然而代码里写的是 pix2pix 必须得用正方形的图= =

然后趁机掌握了 cv2 resize 的各种姿势= =

看 options 里面有很多选项,其实直接用 scripts 文件夹里现成的就好了

要注意的是训练和测试的方向= =

看好自己要 AtoB 还是 BtoA

(以及有个 qa.md 和 tips.md)根本没有链接,藏在代码里头

1
2
3
4
5
# train
python train.py --dataroot ./AB / --name rain --model pix2pix --which_direction AtoB --gpu_ids 0,1 --fineSize 480 --resize_or_crop crop

# test
python test.py --dataroot ./AB --name deraindrop_pix2pix --model pix2pix --which_model_netG unet_256 --which_direction AtoB --dataset_mode aligned --norm batch --resize_or_crop none

上面好像并不对

还有个参数,是控制测试的时候用多少张的,默认是 50 = =,要手动改成自己的才行。


mark 一下尝试跑通了 attentive-GAN

1
CUDA_VISIBLE_DEVICES=0 python test.py --mode demo --input_dir ./100/input/ --output_dir ./100/output/

hmmmm test.py 我自己重写了,大概就是把 demo 和 test 合到一起= =,既存结果又测 psnr。

要占用 7G+ 的显存= =