代码Pix2PixHD: https://github.com/ErinChen1/EPDN
测试集:
15_net_G.pth:
下载解压点击test.py即可运行,如果遇到图像大计算失败,可以单独运行单一图像,或者压缩一下尺寸
一开始并没有执行,是数据集加载错了,而且更改了torch底下的num_workers,应该更改外面的参数
gpu如何关闭
https://blog.csdn.net/xiaosongshine/article/details/89401522
方法一:cuda()->cpu()
方法二:gpu_ids置为空(一共两个文件base_options.py和base_model.py)
后来把cuda()改成了cpu(),主要是models底下的.py文件,把cuda()改成了cpu(),把cuda.Tensor改成了.Tensor,结尾添加#cuda
可气的是转了cpu也不行