图像处理
人像动漫化
百度接口提供的
具体实现
import requests, base64 # 百度AI开放平台鉴权函数 def get_access_token(): url = 'https://aip.baidubce.com/oauth/2.0/token' data = { 'grant_type': 'client_credentials', # 固定值 'client_id': '6M7FeQm****c3p0ex1', # 在开放平台注册后所建应用的API Key 'client_secret': 'XyyUSrFB*****KyBRI0KFuc' # 所建应用的Secret Key } res = requests.post(url, data=data) res = res.json() access_token = res['access_token'] return access_token def image_process(img_before, img_after, how_to_deal): # 函数的三个参数，一个是转化前的文件名，一个是转化后的文件名，均在同一目录下，第三个是图像处理能力选择 request_url = 'https://aip.baidubce.com/rest/2.0/image-process/v1/' + how_to_deal if how_to_deal == 'style_trans': # 判断如果是 图像风格化，需要额外添加一个风格配置 others = 'cartoon' # 风格化参数，具体可设置范围参见下面注释 ''' cartoon：卡通画风格 pencil：铅笔风格 color_pencil：彩色铅笔画风格 warm：彩色糖块油画风格 wave：神奈川冲浪里油画风格 lavender：薰衣草油画风格 mononoke：奇异油画风格 scream：呐喊油画风格 gothic：哥特油画风格 ''' else: others = '' file = open(img_before, 'rb') # 二进制读取图片 origin_img = base64.b64encode(file.read()) # 将图片进行base64编码 headers = {'Content-Type': 'application/x-www-form-urlencoded'} data = { 'access_token': get_access_token(), 'image': origin_img } res = requests.post(request_url, data=data, headers=headers) res = res.json() if res: f = open(img_after, 'wb') after_img = res['image'] after_img = base64.b64decode(after_img) f.write(after_img) f.close() if __name__ == '__main__': img_before = 'img/me.jpg' # 当前目录下的图片 img_after = img_before.split('.') # 将原文件名分成列表 img_after = img_after[0] + '_1.' + img_after[1] # 新生成的文件名为原文件名上加 _1 image_process(img_before, img_after, 'selfie_anime') # 第三个参数： selfie_anime 为人像动漫化，colourize 图像上色，style_trans 为图像风格化 print('done!')
生成结果：
还可以给他戴上口罩
只需在data里加上参数："type":'anime_mask',"mask_id":"2"
import requests, base64 # 百度AI开放平台鉴权函数 def get_access_token(): url = 'https://aip.baidubce.com/oauth/2.0/token' data = { 'grant_type': 'client_credentials', # 固定值 'client_id': '6M7FeQm*****EeDzc3p0ex1', # 在开放平台注册后所建应用的API Key 'client_secret': 'XyyUS****BRI0KFuc' # 所建应用的Secret Key } res = requests.post(url, data=data) res = res.json() access_token = res['access_token'] return access_token def image_process(img_before, img_after, how_to_deal): # 函数的三个参数，一个是转化前的文件名，一个是转化后的文件名，均在同一目录下，第三个是图像处理能力选择 request_url = 'https://aip.baidubce.com/rest/2.0/image-process/v1/' + how_to_deal if how_to_deal == 'style_trans': # 判断如果是 图像风格化，需要额外添加一个风格配置 others = 'cartoon' # 风格化参数，具体可设置范围参见下面注释 ''' cartoon：卡通画风格 pencil：铅笔风格 color_pencil：彩色铅笔画风格 warm：彩色糖块油画风格 wave：神奈川冲浪里油画风格 lavender：薰衣草油画风格 mononoke：奇异油画风格 scream：呐喊油画风格 gothic：哥特油画风格 ''' else: others = '' file = open(img_before, 'rb') # 二进制读取图片 origin_img = base64.b64encode(file.read()) # 将图片进行base64编码 headers = {'Content-Type': 'application/x-www-form-urlencoded'} data = { 'access_token': get_access_token(), 'image': origin_img, "type":'anime_mask', "mask_id":"2" } res = requests.post(request_url, data=data, headers=headers) res = res.json() if res: f = open(img_after, 'wb') after_img = res['image'] after_img = base64.b64decode(after_img) f.write(after_img) f.close() if __name__ == '__main__': img_before = 'img/me.jpg' # 当前目录下的图片 img_after = img_before.split('.') # 将原文件名分成列表 img_after = img_after[0] + '_2.' + img_after[1] # 新生成的文件名为原文件名上加 _1 image_process(img_before, img_after, 'selfie_anime') # 第三个参数： selfie_anime 为人像动漫化，colourize 图像上色，style_trans 为图像风格化 print('done!')
结果如下：
