一起养成写作习惯!这是我参与「掘金日新计划 · 4 月更文挑战」的第23天,点击查看活动详情。
1.由来
最近浏览掘金,看到 腾讯的老照片修复算法,我把它搬到网上,随便玩 ,心有所动,也自己搞个玩玩,不过我用的是飞桨的PaddleHub,代码不多,寥寥十来行,2分钟就实现了。
2.内容
近期从52pojie上下载了一堆老照片,特别喜欢其中的【稀缺珍藏老照片民国34年-38年民生图像】,特对其中的黑白老照片进行了上色处理,碰巧迎接下春天的到来。
3.代码
代码如下,主要就是加载deoldfiy 模型,遍历目标目录,进行上色转换。
#!/usr/bin/python
# -*- coding: UTF-8 -*-
"""
@author:livingbody
@file:modify_pic.py
@time:2022/04/23
"""
import paddlehub as hub
import os
if __name__ == '__main__':
# 加载模型
model = hub.Module(name='deoldify')
# 设置输入图像路径
for parent_path, dir, filenames in os.walk("1947年 陕西"):
for filename in filenames:
print(filename)
print(os.path.join("1947年 陕西", filename))
# 模型预测
result = model.predict(os.path.join("1947年 陕西", filename))
# 获取输出图像路径
output_img = result[1]
print(output_img)
4.效果
4.1 老照片
4.2 上色照片
1947年,陕西。看手相称骨的
1947年,陕西。看相的在掐耳朵
1947年,陕西。看痣相的
1947年,陕西。算卦的
1947年,陕西。渭河平原
1947年,陕西西安。大雁塔
1947年,陕西西安。大雁塔
1947年,陕西西安。大雁塔
1947年,陕西西安。大雁塔
1947年,陕西西安。大雁塔
1947年,陕西西安。大雁塔
1947年,陕西兴平。茂陵
1947年,陕西兴平。茂陵
1947年,陕西兴平。茂陵
1947年,陕西兴平。茂陵
1947年,陕西兴平。茂陵
1947年,中国。几个中国人在路边歇脚,这儿风景不错,有为行人修建的石凳
1947年,中国。几个中国人在路边歇脚,这儿风景不错,有为行人修建的石凳