微软老照片AI修复开源代码运行错误

delcomp

于 2021-02-13 16:00:43 发布

阅读量819

点赞数 2

CC 4.0 BY-SA版权

分类专栏：笔记文章标签：深度学习神经网络 pytorch

本文链接：https://blog.youkuaiyun.com/delcomp/article/details/113801136

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

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

微软老照片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内存都不够。
接着看带划痕的照片

最低0.47元/天解锁文章

200万优质内容无限畅学