百度ai平台接口使用记录

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

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

记录一下操作步骤:

  1. 打开百度ai的平台

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

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

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

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

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

获得AK和SK

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

 

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

获得token

运行下面代码(注意替换成自己的AKSK):

# 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。

我的效果如下:

 

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

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

猜你喜欢

转载自blog.csdn.net/weixin_44492824/article/details/124498215