微软老照片AI修复开源代码运行错误
微软开源代码:https://github.com/microsoft/Bringing-Old-Photos-Back-to-Life
按照说明文件安装必须的第三方包,下载预训练的模型后尝试运行代码:
下面是无划痕照片的修复
python run.py --input_folder [test_image_folder_path]
--output_folder [output_path]
--GPU 0
一切ok,图片稍微大点儿,就用CPU,速度慢一点儿,可已忍受,其实大多数时候都得用cpu运行,多数情况下GPU内存都不够。
接着看带划痕的照片
p

在尝试运行微软开源的老照片AI修复代码时,遇到GPU内存不足及CPU运行时报错的问题。通过分析,确定错误发生在输入类型与模型权重类型不匹配。解决方案是无论是否使用GPU,先将输入数据移到GPU,然后进行CPU运算。经过修改,代码能够正常运行,并展示了修复效果,尤其在人脸增强方面表现出色。然而,对于带有黄斑、霉点的老照片,修复效果并不理想,期待更好的模型或训练数据。
最低0.47元/天 解锁文章
2217

被折叠的 条评论
为什么被折叠?



