最近某音上的动漫特效特别火,很多人都玩着动漫肖像,我媳妇儿也不例外。看着她这么喜欢这个特效,我决定做一个图片处理工具,这样媳妇儿的动漫头像就有着落了。
编码
为了快速实现我们的目标,我们就不自己写图片处理程序了,我们直接调用百度AI开放平台提供的接口实现我们的需求。
这一次我们导入如下两个库
1 #-*- coding:utf-8 -*-
2 #author:菜鸟小白的学习分享
3
4 import requests, base64
我们需要进行百度AI开放平台的接口认证,代码如下:
#-*- coding:utf-8 -*-#author:菜鸟小白的学习分享#百度AI开放平台鉴权函数
defget_access_token():
url= 'https://aip.baidubce.com/oauth/2.0/token'data={'grant_type': 'client_credentials', #固定值
'client_id': 'A3ppUrgl6H*******NjDN4Bb', #在开放平台注册后所建应用的API Key
'client_secret': 'SqaeFpiPPC**********H1lsb0xO3w' #所建应用的Secret Key
}
res= requests.post(url, data=data)
res=res.json()
access_token= res['access_token']return access_token
因为百度AI平台提供了多种图像处理方式,请求URL主体都相同,但是传参不同,所以我们需要对不同的处理方式进行参数处理,代码如下:
1 #-*- coding:utf-8 -*-
2 #author:菜鸟小白的学习分享
3 defget_config():4 img_before = input("请输入当前文件夹下需要处理的图片名称:")5 process_action = ['','selfie_anime','colourize','style_trans']6 print("支持以下处理动作:\n1:为人像动漫化\n2:图像上色\n3:为图像风格化")7 #处理动作: selfie_anime 为人像动漫化,colourize 图像上色,style_trans 为图像风格化
8 i = int(input("请输入需要处理的动作:"))9 """
10 cartoon:卡通画风格11 pencil:铅笔风格12 color_pencil:彩色铅笔画风格13 warm:彩色糖块油画风格14 wave:神奈川冲浪里油画风格15 lavender:薰衣草油画风格16 mononoke:奇异油画风格17 scream:呐喊油画风格18 gothic:哥特油画风格"""
19 others = ['','cartoon','pencil','color_pencil','warm','wave','lavender','mononoke','scream','gothic']20 j =021 if process_action[i] == 'style_trans':22 print("支持转化的风格有:\n\23 1:卡通画风格\n\24 2:铅笔风格\n\25 3:彩色铅笔画风格\n\26 4:彩色糖块油画风格\n\27 5:神奈川冲浪里油画风格\n\28 6:薰衣草油画风格\n\29 7:奇异油画风格\n\30 8:呐喊油画风格\n\31 9:哥特油画风格\n")32 j = int(input("请输入需要转化的风格类型(数字):"))33 return img_before,process_action[i],others[j]
我们获取到图片和处理参数之后,就拿着这些信息去请求百度AI开放平台了
1 #-*- coding:utf-8 -*-
2 #author:菜鸟小白的学习分享
3 defimage_process(img_before, img_after, how_to_deal,others):4 #函数的三个参数,一个是转化前的文件名,一个是转化后的文件名,均在同一目录下,第三个是图像处理能力选择
5 request_url = 'https://aip.baidubce.com/rest/2.0/image-process/v1/' +how_to_deal6 7 file = open(img_before, 'rb') #二进制读取图片
8 origin_img = base64.b64encode(file.read()) #将图片进行base64编码
9 headers = {'Content-Type': 'application/x-www-form-urlencoded'}10 data ={11 'access_token': get_access_token(),12 'image': origin_img,13 'option': others14 }15 16 res = requests.post(request_url, data=data, headers=headers)17 res =res.json()18 19 ifres:20 f = open(img_after, 'wb')21 after_img = res['image']22 after_img =base64.b64decode(after_img)23 f.write(after_img)24 f.close()
最后,我们再把主函数调用写一下
1 #-*- coding:utf-8 -*-
2 #author:菜鸟小白的学习分享
3 if __name__ == '__main__':4 #选择输入信息
5 img_before, process_action, others =get_config()6 img_after = img_before.split('.') #将原文件名分成列表
7 img_after = img_after[0] + '_1.' + img_after[1] #新生成的文件名为原文件名上加 _1
8 image_process(img_before, img_after, process_action,others)9 print('done!')
实现效果
原始图片:
卡通化的转化效果:
彩铅风格转换效果:
其它效果就由小伙伴们自行实验吧,时间太晚了,我就不一一尝试了。
程序获取
考虑到有的小伙伴还没有python环境也想体验工具效果,我把也代码打包成了可执行程序,关注公众号“菜鸟小白的学习分享”回复“图片处理程序”即可获取。
有兴趣自己研究的小伙伴想要获取源码的话,关注公众号“菜鸟小白的学习分享”回复“图片处理源码”即可获取。
如果你也认可菜鸟小白的学习分享的话,请给我一个在看、点赞、关注+赞赏,因为有你们,所以我一直在努力。
# -*- coding:utf-8 -*-#author:菜鸟小白的学习分享if __name__ == '__main__':# 选择输入信息img_before, process_action, others = get_config()img_after = img_before.split('.') # 将原文件名分成列表img_after = img_after[0] + '_1.' + img_after[1] # 新生成的文件名为原文件名上加 _1image_process(img_before, img_after, process_action,others)print('done!')