百度ai平台接口使用记录

  前一段时间在做毕业设计，打算做一些风格转换相关的项目，想自己制作一份数据集用于训练，于是发现了百度ai平台提供的api接口可以实现“人脸动漫化”、“图像风格转换”等功能。按照一定步骤操作后，只要通过post请求就可以将本地的图片进行风格转换了。感觉还不错，比直接使用现有数据集好多了，起码数据质量上可以自己保证~

  当然按照教程来做也非常简单，不过还是做个记录，下次就不用研究教程了。

记录一下操作步骤：

1. 打开百度ai的平台：

  里面有很多功能，我使用的图像转换和人脸动漫化（其实都差不多），选择图像风格转换进入。

  接下来可以看到各种提供的图像风格转换的效果，我们直接使用，点开“产品价格”，往下可以看到有免费几百次的使用机会，点击“立即使用”。

  现在来到了“百度智能云”界面，先注册一个账号，并登录，来到主页：

  先“领取免费资源”，进去后选择你想要的尝试的功能，我选了“图像转换”和“人脸动漫化”之类的功能。（反正也都是免费选，大不了全选了）

  然后出来之后“创建应用”（我这里已经有一个了），名称和描述都随意，把需要的功能选上就行（我选了图像转换、人脸动漫化等）。

获得AK和SK

点击“管理应用”，可以看到自己的AK和SK

 

拿到这两个字符串后，下一步获取token。

获得token

运行下面代码（注意替换成自己的AK和SK）：

# encoding:utf-8
import requests

# client_id 为官网获取的AK， client_secret 为官网获取的SK
host = 'https://aip.baidubce.com/oauth/2.0/token?grant_type=client_credentials&client_id=【官网获取的AK】&client_secret=【官网获取的SK】'

response = requests.get(host)
if response:
    print(response.json())

注意替换代码里的AK和SK。

运行后会得到一个json串，里面的refresh_token字段就是我们的token。把它复制保存起来，我们马上就要用。

有了token就可以使用post请求进行图像风格转换测试了：

模板：

# encoding:utf-8
import requests
import base64

'''
图像风格转换
'''

request_url = "https://aip.baidubce.com/rest/2.0/image-process/v1/style_trans"
# 二进制方式打开图片文件
f = open('./test.png', 'rb')
img = base64.b64encode(f.read())
#这个option是一些官网提供的风格选项，我最喜欢mononoke风格~
# params = {"image":img,"option":"cartoon"}
# params = {"image":img,"option":"color_pencil"}
# params = {"image":img,"option":"warm"}
params = {"image":img,"option":"mononoke"}
# params = {"image":img,"option":"gothic"}
# params = {"image":img,"option":"lavender"}

access_token = '你复制的token'

request_url = request_url + "?access_token=" + access_token
headers = {'content-type': 'application/x-www-form-urlencoded'}
response = requests.post(request_url, data=params, headers=headers)
if response:
    result = response.json()
    print(type(result))
    print(result)

注意替换代码里的token。这里test.png图片可以自己随便选一张照片放在相同目录下，运行之后，就会在相同目录下得到结果res_mononoke.png。

我的效果如下：

 

  感觉这个风格还是蛮帅的，后面风格转换也打算以此作为参照来生成数据集，一般购买一千次调用次数应该就够用于训练了。

  将我这个例子玩过之后，就可以去官网尝试其他功能了，也都差不多是这样使用~