图像增强论文复现

 

 

 

 

 

代码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也不行