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

最新推荐文章于 2024-04-19 09:34:42 发布

you_wait_I_come

最新推荐文章于 2024-04-19 09:34:42 发布

阅读量3.5k

点赞数 4

文章标签： python 人工智能百度

本文链接：https://blog.csdn.net/you_wait_I_come/article/details/107023122

版权

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_token
    headers = {
        '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_image
    else:
        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、BMP
        if os.path.splitext(image_name)[1] in image_type:
            image_path = file_path + image_name
            image_paths.append(image_path)
    if image_paths:
        return image_paths
    else:
        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 image
    except 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、BMP
        if os.path.splitext(image_name)[1] in image_type:
            image_path = file_path + image_name
            image_paths.append(image_path)
    if image_paths:
        return image_paths
    else:
        print('原源文件夹没有找到相关格式的图片！')


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


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失败！')


def load_image(image_path):
    '''加载指定路径的图片'''
    try:
        with open(image_path, 'rb') as f:
            image = base64.b64encode(f.read())
        return image
    except 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_token
    headers = {
        '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_image
    else:
        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相关的信息，请移步（后面补上）。

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

you_wait_I_come

关注

4
点赞
踩
26

收藏

觉得还不错? 一键收藏
7
评论
Python：批量增强图片清晰度 ——基于百度API，含获取的AK、 SK的流程，access_token的获取流程，以及生成.exe文件

Python：批量增强图片清晰度 ——基于百度API，含获取的AK、 SK的流程，access_token的获取流程最近需要提高一些图片的清晰度，但一张张的去弄太费时间了，所以打算尝试用Python来简化流程。涉及关键词与模块：百度API相关：获取AK、SK值，获取access token 值爬虫相关：requests库操作系统：os库编码：base64库以下是完整的流程，或许对弟兄姐妹们有帮助。流程（想自己编程尝试的可以看看）一. 获取 access_token 值（百度API相关）
复制链接

扫一扫