图像变换风格
import requests, base64 # 百度AI开放平台鉴权函数 def get_access_token(): url = 'https://aip.baidubce.com/oauth/2.0/token' data = { 'grant_type': 'client_credentials', # 固定值 'client_id': '****', # 在开放平台注册后所建应用的API Key 'client_secret': '*******' # 所建应用的Secret Key } res = requests.post(url, data=data) res = res.json() access_token = res['access_token'] return access_token def get_config(): img_before = input("请输入当前文件夹下需要处理的图片名称:") process_action = ['','selfie_anime','colourize','style_trans'] print("支持以下处理动作:\n1:为人像动漫化\n2:图像上色\n3:为图像风格化") # 处理动作: selfie_anime 为人像动漫化,colourize 图像上色,style_trans 为图像风格化 i = int(input("请输入需要处理的动作:")) """ cartoon:卡通画风格 pencil:铅笔风格 color_pencil:彩色铅笔画风格 warm:彩色糖块油画风格 wave:神奈川冲浪里油画风格 lavender:薰衣草油画风格 mononoke:奇异油画风格 scream:呐喊油画风格 gothic:哥特油画风格""" others = ['','cartoon','pencil','color_pencil','warm','wave','lavender','mononoke','scream','gothic'] j = 0 if process_action[i] == 'style_trans': print("支持转化的风格有:\n\ 1:卡通画风格\n\ 2:铅笔风格\n\ 3:彩色铅笔画风格\n\ 4:彩色糖块油画风格\n\ 5:神奈川冲浪里油画风格\n\ 6:薰衣草油画风格\n\ 7:奇异油画风格\n\ 8:呐喊油画风格\n\ 9:哥特油画风格\n") j = int(input("请输入需要转化的风格类型(数字):")) return img_before,process_action[i],others[j] def image_process(img_before, img_after, how_to_deal,others): # 函数的三个参数,一个是转化前的文件名,一个是转化后的文件名,均在同一目录下,第三个是图像处理能力选择 request_url = 'https://aip.baidubce.com/rest/2.0/image-process/v1/' + how_to_deal 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, 'option': others } 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, process_action, others = get_config() img_after = img_before.split('.') # 将原文件名分成列表 img_after = img_after[0] + '_1.' + img_after[1] # 新生成的文件名为原文件名上加 _1 image_process(img_before, img_after, process_action,others) print('done!')
-----------------------------------------------------------------------------------------------------------------------------------------