Python 调用图像融合API

时间:2022-05-01 16:10:38

Python 调用图像融合API

本文记录使用Python,调用腾讯AI开放平台的图像融合API。官网给出的Demo用的是PHP,博主作为Python的粉丝,自然想用它来和『最好的』的语言一较高下,顺便加深对服务调用的理解

官网PHP实现

腾讯的官方文档应该写的非常详细了,可以直接运行的PHP代码如下:

<?php

// getReqSign :根据 接口请求参数 和 应用密钥 计算 请求签名
// 参数说明
// - $params:接口请求参数(特别注意:不同的接口,参数对一般不一样,请以具体接口要求为准)
// - $appkey:应用密钥
// 返回数据
// - 签名结果
function getReqSign($params /* 关联数组 */, $appkey /* 字符串*/)
{
// 1. 字典升序排序
ksort($params); // 2. 拼按URL键值对
$str = '';
foreach ($params as $key => $value)
{
if ($value !== '')
{
$str .= $key . '=' . urlencode($value) . '&';
}
} // 3. 拼接app_key
$str .= 'app_key=' . $appkey; // 4. MD5运算+转换大写,得到请求签名
$sign = strtoupper(md5($str));
return $sign;
} // doHttpPost :执行POST请求,并取回响应结果
// 参数说明
// - $url :接口请求地址
// - $params:完整接口请求参数(特别注意:不同的接口,参数对一般不一样,请以具体接口要求为准)
// 返回数据
// - 返回false表示失败,否则表示API成功返回的HTTP BODY部分
function doHttpPost($url, $params)
{
$curl = curl_init(); $response = false;
do
{
// 1. 设置HTTP URL (API地址)
curl_setopt($curl, CURLOPT_URL, $url); // 2. 设置HTTP HEADER (表单POST)
$head = array(
'Content-Type: application/x-www-form-urlencoded'
);
curl_setopt($curl, CURLOPT_HTTPHEADER, $head); // 3. 设置HTTP BODY (URL键值对)
$body = http_build_query($params);
curl_setopt($curl, CURLOPT_POST, true);
curl_setopt($curl, CURLOPT_POSTFIELDS, $body); // 4. 调用API,获取响应结果
curl_setopt($curl, CURLOPT_HEADER, false);
curl_setopt($curl, CURLOPT_NOBODY, false);
curl_setopt($curl, CURLOPT_RETURNTRANSFER, true);
curl_setopt($curl, CURLOPT_SSL_VERIFYHOST, true);
curl_setopt($curl, CURLOPT_SSL_VERIFYPEER, false);
$response = curl_exec($curl);
if ($response === false)
{
$response = false;
break;
} $code = curl_getinfo($curl, CURLINFO_HTTP_CODE);
if ($code != 200)
{
$response = false;
break;
}
} while (0); curl_close($curl);
return $response;
} // 图片base64编码
$path = 'test.jpg';
$data = file_get_contents($path);
$base64 = base64_encode($data); // 设置请求数据
$appkey = 'your_appkey';
$params = array(
'app_id' => 'your_app_id',
'image' => $base64,
'model' => '1',
'time_stamp' => strval(time()),
'nonce_str' => strval(rand()),
'sign' => '',
);
$params['sign'] = getReqSign($params, $appkey); // 执行API调用
$url = 'https://api.ai.qq.com/fcgi-bin/ptu/ptu_facemerge';
$response = doHttpPost($url, $params);
echo $response;

根据此程序,可以返回融合图片的base64编码,再解码就可以得到图片

Python版实现

以下代码是博主参照php的改写而成,值得注意的问题是urlencode是一种特定的数据格式,尤其是图片信息一定要按此编码,否则返回错误信息

# coding=utf-8
import requests
import time
import random
import hashlib
import base64
from urllib import urlencode def get_sign(para, app_key):
# 签名的key有严格要求,按照key升序排列
data = sorted(para.items(), key=lambda item: item[0])
s = urlencode(data)
# app_key最后加
s += '&app_key=' + app_key
# 计算md5报文信息
md5 = hashlib.md5()
md5.update(s)
digest = md5.hexdigest()
return digest.upper() url = 'https://api.ai.qq.com/fcgi-bin/ptu/ptu_facemerge'
app_key = 'your_app_key'
# 读取图片数据
raw_data = open('test.jpg').read()
image_data = base64.b64encode(raw_data) # 定义发送的post数据
data = {
'app_id': 'your_app_id',
'image': image_data,
'model': '1', # 选定想要融合的模板
'time_stamp': str(int(time.time())),
'nonce_str': str(random.random()),
} data['sign'] = get_sign(data, app_key) # 发送post请求
r = requests.post(url, data=data)
# 将得到的数据解码,然后保存到jpg
res_image = base64.b64decode(r.json()['data']['image'])
img = open('out.jpg', 'w')
img.write(res_image)
img.close()

效果

输入图片(网上的人脸图片,侵删)

test.jpg

Python 调用图像融合API

腾讯给出的融合模板1

Python 调用图像融合API

输出

out.jpg

Python 调用图像融合API

小结

这次实验的逻辑其实不是很复杂,知识点主要是签名算法POST请求。可以看到Python的代码比PHP简短得多,主要是用到了一些封装好的库

有了人脸融合这项服务后,感觉可以做很多有趣的事情~