Python+Baidu.Api实现灰度(BW)图像自上色

作为深度学习领域的现今研究热门,灰度图像再着色还是比较有意义的,至少在一些年代比较久远的照片、视频彩色复现上应用意义比较明显!

1.申请一个百度智能云开发者账号并实名个人认证

选择图像增强应用,创建一个灰度图像上色应用,并注意该应用的三个核心参数:

APP_ID = 'XXX'
API_KEY = 'XXXXXXXXXXXXXXXX'
SECRET_KEY = 'XXXXXXXXXXXXXXXXXXXXXXX'

领取免费资源包,注意,应用创建后并不能直接使用(需付费),但是平台提供了1000次的免费测试机会(365天范围内),只需申领一下即可,注意,只有实名认证之后方可申领!

2.编辑Python代码

# 2022.03.30
# print('Hello,world\n')


from aip import AipImageProcess
from PIL import Image
import base64



APP_ID = ''
API_KEY = ''
SECRET_KEY = ''

client = AipImageProcess(APP_ID, API_KEY, SECRET_KEY)

def get_file_content(filePath):
    with open(filePath, 'rb') as fp:
        return fp.read()

image = get_file_content('IMG_20180925_111155.jpg')

# client.colourize(image);

result = client.colourize(image);
print(type(result))
# print(result)


img = base64.b64decode(result['image'])
file = open('one.jpg','wb')
file.write(img)
file.close()


'''
im1 = Image.open('tmp1f51e5jo.png')
im1.show()

im2 = Image.open('ColorRem.jpg')
im2.show()
'''

当然,注意需要安装对应的Package,直接pip即可,缺啥装啥,唯一需要注意的是,百度aip的安装包全称:baidu-aip!

3.测试图片上色效果(实际图片+卡通图片)

卡通图片着色效果差异还是很大的,实际图片差异也比较大,不过在蓝天白云上的细节处理相当到位,草色田野什么的就有些失真了!

附录:代码包(Direct)

发表回复

您的电子邮箱地址不会被公开。 必填项已用*标注


皖ICP备2021003932号
召唤伊斯特瓦尔