图像增强论文复现

 

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

原文地址:https://www.cnblogs.com/2008nmj/p/13503662.html