Python：批量增强图片清晰度 ——基于百度API，含获取的AK、 SK的流程，access_token的获取流程，以及生成.exe文件

最近需要提高一些图片的清晰度，但一张张的去弄太费时间了，所以打算尝试用Python来简化流程。

涉及关键词与模块：
百度API相关：获取AK、SK值，获取access token 值
爬虫相关：requests库
操作系统：os库
编码：base64库
.exe文件：pyinstaller库

以下是完整的流程，或许对弟兄姐妹们有帮助。

流程（想自己编程尝试的可以看看）

一. 获取 access_token 值（百度API相关）

找到相关接口文档
先看看百度AI平台，点击左上角的开放功能、选择图像技术里面的图像效果增强，就可以看到图片清晰度增强了。点击后找到蓝色字体的**接口文档**，点开就可以阅读相关的接口文档信息。
获取 AK，SK 值
通过阅读文档我们知道要获取参数access_token的值，必须有API Key(AK)、Secret Key(SK), 获取AK、SK需要登录百度智能云账号（没有就注册一个）。鼠标移动到左边黑色框里面的箭头，出现产品栏。鼠标移动到人工智能，选择图片效果增强，在概览界面选择管理应用(如果没有应用就建立一个)。进入应用列表后就可以看到API Key、Secret Key（点显示即可）。于是就获得了AK、SK了
获取 access_token 值
回到**接口文档，在URL参数栏选择Access Token获取阅读可知，获取Access Token值需要访问url = https://aip.baidubce.com/oauth/2.0/token?grant_type=client_credentials&client_id=【您的AK】&client_secret=【您的SK】。通过post请求访问这个网址就能得到access_token**的值。

编程获取 access token
代码如下：

def get_access_token(ak, sk):'''获取access token'''host = 'https://aip.baidubce.com/oauth/2.0/token?grant_type=client_credentials&client_id={0}&client_secret={1}'.format(ak, sk)response = requests.post(host).json()if response:return response['access_token']else:print('获取Access Token失败！')

注意：
返回的的 json 文本参数是一个字典。

二. 获得增强清晰度的图片（爬虫相关）

继续阅读**接口文档**可以获得相关的 request_url、headers、params、response的相关信息, 通过post请求就可以获得增强了清晰度的图片。

另外，想了解爬虫的相关信息，请移步到我的其他博客（后面补上）

编程获取增强了清晰度的图片
代码如下：

def enhance_image( ak, sk, old_image):'''增强图片清晰度'''access_token = get_access_token(ak, sk)url = "https://aip.baidubce.com/rest/2.0/image-process/v1/image_definition_enhance"request_url = url + "?access_token=" + access_tokenheaders = {'content-type': 'application/x-www-form-urlencoded'}params = {"image": old_image}response = requests.post(request_url, data=params, headers=headers).json()if 'error_msg' not in response:     # 通过返回信息判断是否符合要求new_image = response['image']return new_imageelse:print('增强图片失败！图片大小不能超过2M!\t')

三. 加载、存储图片（操作系统、编码相关）

下面解决将符合格式的图片加载到python，又将处理后的图片正确命名。其代码如下：

另外，想了解关于操作系统及编码的相关信息，请移步到我的其他博客（后面补上）

获得图片路径

def get_image_paths(file_path):'''得到全部符合格式的图片的路径'''image_type = ['.png', '.jpg', '.jpeg', '.bmp', '.PNG', '.JPG', '.JPEG', '.BMP']image_names = os.listdir(file_path)image_paths = []for image_name in image_names:      # 只支持PNG、JPG、JPEG、BMPif os.path.splitext(image_name)[1] in image_type:image_path = file_path + image_nameimage_paths.append(image_path)if image_paths:return image_pathselse:print('原源文件夹没有找到相关格式的图片！')

获得图片名称

def get_image_name(image_path):'''获取图片的名称'''image_name = image_path.split('\\')[-1]return image_name

加载待处理图片

def load_image(image_path):'''加载指定路径的图片'''try:with open(image_path, 'rb') as f:image = base64.b64encode(f.read())return imageexcept Exception as e:print('加载文件路径失败，请检查输入的路径\n', e)

存储处理后的图片

def save_image(image_path, image):'''存储图片到指定文件夹'''try:image = base64.b64decode(image)with open(image_path, 'wb') as f:f.write(image)except Exception as e:print('存储图片失败！\n', e)

四. 完整代码

其实，仔细想一下就知道，在这些代码的基础上，可以将百度API的很多功能整合在一起，整合成一个库。笔者没有去查是否已经有这个库了，不过后面有时间会将其完善。有兴趣的朋友可以尝试一下哦。

# encoding:utf-8
import requests
import base64
import os'''
图像清晰度增强
'''# client_id 为官网获取的AK， client_secret 为官网获取的SK
def get_image_paths(file_path):'''得到全部符合格式的图片的路径'''image_type = ['.png', '.jpg', '.jpeg', '.bmp', '.PNG', '.JPG', '.JPEG', '.BMP']image_names = os.listdir(file_path)image_paths = []for image_name in image_names:      # 只支持PNG、JPG、JPEG、BMPif os.path.splitext(image_name)[1] in image_type:image_path = file_path + image_nameimage_paths.append(image_path)if image_paths:return image_pathselse:print('原源文件夹没有找到相关格式的图片！')def get_image_name(image_path):'''获取图片的名称'''image_name = image_path.split('\\')[-1]return image_namedef get_access_token(ak, sk):'''获取access token'''host = 'https://aip.baidubce.com/oauth/2.0/token?grant_type=client_credentials&client_id={0}&client_secret={1}'.format(ak, sk)response = requests.post(host).json()if response:return response['access_token']else:print('获取Access Token失败！')def load_image(image_path):'''加载指定路径的图片'''try:with open(image_path, 'rb') as f:image = base64.b64encode(f.read())return imageexcept Exception as e:print('加载文件路径失败，请检查输入的路径\n', e)def enhance_image( ak, sk, old_image):'''增强图片清晰度'''access_token = get_access_token(ak, sk)url = "https://aip.baidubce.com/rest/2.0/image-process/v1/image_definition_enhance"request_url = url + "?access_token=" + access_tokenheaders = {'content-type': 'application/x-www-form-urlencoded'}params = {"image": old_image}response = requests.post(request_url, data=params, headers=headers).json()if 'error_msg' not in response:     # 通过返回信息判断是否符合要求new_image = response['image']return new_imageelse:print('增强图片失败！图片大小不能超过2M!\t')def save_image(image_path, image):'''存储图片到指定文件夹'''try:image = base64.b64decode(image)with open(image_path, 'wb') as f:f.write(image)except Exception as e:print('存储图片失败！\n', e)if __name__ == '__main__':AK = '您的AK'SK = '您的SK'Old_file_path = input('请输入要处理的文件存放目录：') + '\\'New_file_path = input('请输入处理后的文件存放目录：') + '\\'Old_image_paths = get_image_paths(Old_file_path)if Old_image_paths:print("正在处理中...")for Old_image_path in Old_image_paths:New_image_path = New_file_path + get_image_name(Old_image_path)Old_image = load_image(Old_image_path)New_image = enhance_image(AK, SK, Old_image)save_image(New_image_path, New_image)print("处理完成！")## 码字不易，求朋友关注一下呦

五. 生成.exe文件（pyinstaller库）

安装好pyinstaller库后，在**pycharm终端（terminal）**中输入：

 pyinstaller -F 文件名.py

就可以生成.exe 文件啦，其位置在文件名.py 文件夹中。

另外，和pyinstaller相关的信息，请移步（后面补上）。

如果存在问题请朋友批评指正