关于输出的图片全为黑色的问题 · Issue #330 · xinntao/Real

您所在的位置:网站首页 全黑的图片上有字图片 关于输出的图片全为黑色的问题 · Issue #330 · xinntao/Real

关于输出的图片全为黑色的问题 · Issue #330 · xinntao/Real

2024-07-11 19:33| 来源: 网络整理| 查看: 265

@gioia010321 非常感谢您的提醒,我测试了,RealESRGAN项目示例命令加上--fp32后,果然输出正常了!

在GFPGAN项目中使用时,需要修改inference_gfpgan.py的76行,half=False,再用示例命令python inference_gfpgan.py -i inputs/whole_imgs -o results -v 1.3 -s 2,restored_imgs里的图片输出正常,脸部得到优化,背景也得到了优化。

但遇到一点疑惑,在修改inference_gfpgan.py的76行时,注释提示:# need to set False in CPU mode,难道我修改后之所以成功,是使用了CPU,没有使用CUDA? 我又进一步阅读了RealESRGAN项目的示例代码,Real-ESRGAN/realesrgan/utils.py第38行,是判断是否支持CUDA以设置pytorch的设备,print(self.device),运行后输出为:cuda,表示自动选择使用了cuda。我手动将device设置为‘cpu’,再次运行测试代码,明显感觉速度慢很多,证明cuda是生效了的。那么,RealESRGANer的half参数,究竟是什么作用呢?只是决定运算精度,还是决定cpu/cuda模式?

half设置为False后,目前一切正常。

再次感谢 @gioia010321 ,感谢作者!



【本文地址】


今日新闻


推荐新闻


CopyRight 2018-2019 办公设备维修网 版权所有 豫ICP备15022753号-3