资讯专栏INFORMATION COLUMN

Python公众号开发(二)—颜值检测

Code4App / 2539人阅读

摘要:上篇文章,我们把自己的程序接入了微信公众号,并且能把用户发送的文本及图片文件原样返回。微信的机制,我们的程序必须在内给出响应。上篇微信公众号开发小白篇一

上篇文章,我们把自己的程序接入了微信公众号,并且能把用户发送的文本及图片文件原样返回。今天我们把用户的图片通过腾讯的AI平台分析后再返回给用户。

为了防止我的文章被到处转载,贴一下我的公众号【智能制造社区】,欢迎大家关注。

github仓库地址https://github.com/injetlee/Python/tree/master/wechat

效果图

一. 接入腾讯AI平台

我们先看一下官方人脸检测与分析接口的描述:

检测给定图片(Image)中的所有人脸(Face)的位置和相应的面部属性。位置包括(x, y, w, h),面部属性包括性别(gender), 年龄(age), 表情(expression), 魅力(beauty), 眼镜(glass)和姿态(pitch,roll,yaw)。

请求参数包括下面几个:

app_id 应用标识,我们在AI平台注册后就可以得到app_id

time_stamp 时间戳

nonce_str 随机字符串

sign 签名信息,需要我们自己去计算

image 需要检测的图片(上限1M)

mode 检测模式

1.接口鉴权,构造请求参数

官方给了我们接口鉴权的计算方法。

请求参数对按key进行字典升序排序,得到有序的参数对列表N

将列表N中的参数对按URL键值对的格式拼接成字符串,得到字符串T(如:key1=value1&key2=value2),URL键值拼接过程value部分需要URL编码,URL编码算法用大写字母,例如%E8,而不是小写%e8

将应用密钥以app_key为键名,组成URL键值拼接到字符串T末尾,得到字符串S(如:key1=value1&key2=value2&app_key=密钥)

对字符串S进行MD5运算,将得到的MD5值所有字符转换成大写,得到接口请求签名

2.请求接口地址

请求接口信息,我们用 requests 发送请求,会得到返回的 json 格式的图像信息pip install requests安装requests。

3.处理返回的信息

处理返回的信息,把信息展示在图片上,再把处理后的图片保存。这里我们用到 opencv ,和 pillow 两个库pip install pillowpip install opencv-python来安装。

开始编写代码,我们新建一个face_id.py 文件来对接AI平台,并且返回检测后的图像数据。

</>复制代码

  1. import time
  2. import random
  3. import base64
  4. import hashlib
  5. import requests
  6. from urllib.parse import urlencode
  7. import cv2
  8. import numpy as np
  9. from PIL import Image, ImageDraw, ImageFont
  10. import os
  11. # 一.计算接口鉴权,构造请求参数
  12. def random_str():
  13. """得到随机字符串nonce_str"""
  14. str = "abcdefghijklmnopqrstuvwxyz"
  15. r = ""
  16. for i in range(15):
  17. index = random.randint(0,25)
  18. r += str[index]
  19. return r
  20. def image(name):
  21. with open(name, "rb") as f:
  22. content = f.read()
  23. return base64.b64encode(content)
  24. def get_params(img):
  25. """组织接口请求的参数形式,并且计算sign接口鉴权信息,
  26. 最终返回接口请求所需要的参数字典"""
  27. params = {
  28. "app_id": "1106860829",
  29. "time_stamp": str(int(time.time())),
  30. "nonce_str": random_str(),
  31. "image": img,
  32. "mode": "0"
  33. }
  34. sort_dict = sorted(params.items(), key=lambda item: item[0], reverse=False) # 排序
  35. sort_dict.append(("app_key", "P8Gt8nxi6k8vLKbS")) # 添加app_key
  36. rawtext = urlencode(sort_dict).encode() # URL编码
  37. sha = hashlib.md5()
  38. sha.update(rawtext)
  39. md5text = sha.hexdigest().upper() # 计算出sign,接口鉴权
  40. params["sign"] = md5text # 添加到请求参数列表中
  41. return params
  42. # 二.请求接口URL
  43. def access_api(img):
  44. frame = cv2.imread(img)
  45. nparry_encode = cv2.imencode(".jpg", frame)[1]
  46. data_encode = np.array(nparry_encode)
  47. img_encode = base64.b64encode(data_encode) # 图片转为base64编码格式
  48. url = "https://api.ai.qq.com/fcgi-bin/face/face_detectface"
  49. res = requests.post(url, get_params(img_encode)).json() # 请求URL,得到json信息
  50. # 把信息显示到图片上
  51. if res["ret"] == 0: # 0代表请求成功
  52. pil_img = Image.fromarray(cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)) # 把opencv格式转换为PIL格式,方便写汉字
  53. draw = ImageDraw.Draw(pil_img)
  54. for obj in res["data"]["face_list"]:
  55. img_width = res["data"]["image_width"] # 图像宽度
  56. img_height = res["data"]["image_height"] # 图像高度
  57. # print(obj)
  58. x = obj["x"] # 人脸框左上角x坐标
  59. y = obj["y"] # 人脸框左上角y坐标
  60. w = obj["width"] # 人脸框宽度
  61. h = obj["height"] # 人脸框高度
  62. # 根据返回的值,自定义一下显示的文字内容
  63. if obj["glass"] == 1: # 眼镜
  64. glass = "有"
  65. else:
  66. glass = "无"
  67. if obj["gender"] >= 70: # 性别值从0-100表示从女性到男性
  68. gender = "男"
  69. elif 50 <= obj["gender"] < 70:
  70. gender = "娘"
  71. elif obj["gender"] < 30:
  72. gender = "女"
  73. else:
  74. gender = "女汉子"
  75. if 90 < obj["expression"] <= 100: # 表情从0-100,表示笑的程度
  76. expression = "一笑倾城"
  77. elif 80 < obj["expression"] <= 90:
  78. expression = "心花怒放"
  79. elif 70 < obj["expression"] <= 80:
  80. expression = "兴高采烈"
  81. elif 60 < obj["expression"] <= 70:
  82. expression = "眉开眼笑"
  83. elif 50 < obj["expression"] <= 60:
  84. expression = "喜上眉梢"
  85. elif 40 < obj["expression"] <= 50:
  86. expression = "喜气洋洋"
  87. elif 30 < obj["expression"] <= 40:
  88. expression = "笑逐颜开"
  89. elif 20 < obj["expression"] <= 30:
  90. expression = "似笑非笑"
  91. elif 10 < obj["expression"] <= 20:
  92. expression = "半嗔半喜"
  93. elif 0 <= obj["expression"] <= 10:
  94. expression = "黯然伤神"
  95. delt = h // 5 # 确定文字垂直距离
  96. # 写入图片
  97. if len(res["data"]["face_list"]) > 1: # 检测到多个人脸,就把信息写入人脸框内
  98. font = ImageFont.truetype("yahei.ttf", w // 8, encoding="utf-8") # 提前把字体文件下载好
  99. draw.text((x + 10, y + 10), "性别 :" + gender, (76, 176, 80), font=font)
  100. draw.text((x + 10, y + 10 + delt * 1), "年龄 :" + str(obj["age"]), (76, 176, 80), font=font)
  101. draw.text((x + 10, y + 10 + delt * 2), "表情 :" + expression, (76, 176, 80), font=font)
  102. draw.text((x + 10, y + 10 + delt * 3), "魅力 :" + str(obj["beauty"]), (76, 176, 80), font=font)
  103. draw.text((x + 10, y + 10 + delt * 4), "眼镜 :" + glass, (76, 176, 80), font=font)
  104. elif img_width - x - w < 170: # 避免图片太窄,导致文字显示不完全
  105. font = ImageFont.truetype("yahei.ttf", w // 8, encoding="utf-8")
  106. draw.text((x + 10, y + 10), "性别 :" + gender, (76, 176, 80), font=font)
  107. draw.text((x + 10, y + 10 + delt * 1), "年龄 :" + str(obj["age"]), (76, 176, 80), font=font)
  108. draw.text((x + 10, y + 10 + delt * 2), "表情 :" + expression, (76, 176, 80), font=font)
  109. draw.text((x + 10, y + 10 + delt * 3), "魅力 :" + str(obj["beauty"]), (76, 176, 80), font=font)
  110. draw.text((x + 10, y + 10 + delt * 4), "眼镜 :" + glass, (76, 176, 80), font=font)
  111. else:
  112. font = ImageFont.truetype("yahei.ttf", 20, encoding="utf-8")
  113. draw.text((x + w + 10, y + 10), "性别 :" + gender, (76, 176, 80), font=font)
  114. draw.text((x + w + 10, y + 10 + delt * 1), "年龄 :" + str(obj["age"]), (76, 176, 80), font=font)
  115. draw.text((x + w + 10, y + 10 + delt * 2), "表情 :" + expression, (76, 176, 80), font=font)
  116. draw.text((x + w + 10, y + 10 + delt * 3), "魅力 :" + str(obj["beauty"]), (76, 176, 80), font=font)
  117. draw.text((x + w + 10, y + 10 + delt * 4), "眼镜 :" + glass, (76, 176, 80), font=font)
  118. draw.rectangle((x, y, x + w, y + h), outline="#4CB050") # 画出人脸方框
  119. cv2img = cv2.cvtColor(np.array(pil_img), cv2.COLOR_RGB2BGR) # 把 pil 格式转换为 cv
  120. cv2.imwrite("faces/{}".format(os.path.basename(img)), cv2img) # 保存图片到 face 文件夹下
  121. return "检测成功"
  122. else:
  123. return "检测失败"

到这里我们的人脸检测接口接入及图片处理就完成了。之后在收到用户发送的图片信息后,调用这个函数,把处理后的图片返回给用户就可以。

返回图片给用户

当收到用户图片时,需要以下几个步骤:

保存图片

当接收到用户图片后,我们要先把图片保存起来,之后才能去调用人脸分析接口,把图片信息传递过去,我们需要编写一个 img_download 函数来下载图片。详见下方代码

调用人脸分析接口

图片下载后,调用 face_id.py 文件里的接口函数,得到处理后的图片。

上传图片

检测结果是一张新的图片,要把图片发送给用户我们需要一个 Media_ID,要获取Media_ID必须先把图片上传为临时素材,所以这里我们需要一个img_upload函数来上传图片,并且在上传时需要用到一个access_token,我们通过一个函数来获取. 获取access_token必须要把我们自己的IP地址加入白名单,否则是获取不到的。请登录“微信公众平台-开发-基本配置”提前将服务器IP地址添加到IP白名单中,可以在http://ip.qq.com/查看本机的IP...

开始编写代码,我们新建一个 utils.py 来下载、上传图片

</>复制代码

  1. import requests
  2. import json
  3. import threading
  4. import time
  5. import os
  6. token = ""
  7. app_id = "wxfc6adcdd7593a712"
  8. secret = "429d85da0244792be19e0deb29615128"
  9. def img_download(url, name):
  10. r = requests.get(url)
  11. with open("images/{}-{}.jpg".format(name, time.strftime("%Y_%m_%d%H_%M_%S", time.localtime())), "wb") as fd:
  12. fd.write(r.content)
  13. if os.path.getsize(fd.name) >= 1048576:
  14. return "large"
  15. # print("namename", os.path.basename(fd.name))
  16. return os.path.basename(fd.name)
  17. def get_access_token(appid, secret):
  18. """获取access_token,100分钟刷新一次"""
  19. url = "https://api.weixin.qq.com/cgi-bin/token?grant_type=client_credential&appid={}&secret={}".format(appid, secret)
  20. r = requests.get(url)
  21. parse_json = json.loads(r.text)
  22. global token
  23. token = parse_json["access_token"]
  24. global timer
  25. timer = threading.Timer(6000, get_access_token)
  26. timer.start()
  27. def img_upload(mediaType, name):
  28. global token
  29. url = "https://api.weixin.qq.com/cgi-bin/media/upload?access_token=%s&type=%s" % (token, mediaType)
  30. files = {"media": open("{}".format(name), "rb")}
  31. r = requests.post(url, files=files)
  32. parse_json = json.loads(r.text)
  33. return parse_json["media_id"]
  34. get_access_token(app_id, secret)
返回给用户

我们简单修改下收到图片后的逻辑,收到图片后经过人脸检测,上传获得Media_ID,我们要做的就是把图片返回给用户即可。直接看connect.py的代码

</>复制代码

  1. import falcon
  2. from falcon import uri
  3. from wechatpy.utils import check_signature
  4. from wechatpy.exceptions import InvalidSignatureException
  5. from wechatpy import parse_message
  6. from wechatpy.replies import TextReply, ImageReply
  7. from utils import img_download, img_upload
  8. from face_id import access_api
  9. class Connect(object):
  10. def on_get(self, req, resp):
  11. query_string = req.query_string
  12. query_list = query_string.split("&")
  13. b = {}
  14. for i in query_list:
  15. b[i.split("=")[0]] = i.split("=")[1]
  16. try:
  17. check_signature(token="lengxiao", signature=b["signature"], timestamp=b["timestamp"], nonce=b["nonce"])
  18. resp.body = (b["echostr"])
  19. except InvalidSignatureException:
  20. pass
  21. resp.status = falcon.HTTP_200
  22. def on_post(self, req, resp):
  23. xml = req.stream.read()
  24. msg = parse_message(xml)
  25. if msg.type == "text":
  26. reply = TextReply(content=msg.content, message=msg)
  27. xml = reply.render()
  28. resp.body = (xml)
  29. resp.status = falcon.HTTP_200
  30. elif msg.type == "image":
  31. name = img_download(msg.image, msg.source) # 下载图片
  32. r = access_api("images/" + name)
  33. if r == "检测成功":
  34. media_id = img_upload("image", "faces/" + name) # 上传图片,得到 media_id
  35. reply = ImageReply(media_id=media_id, message=msg)
  36. else:
  37. reply = TextReply(content="人脸检测失败,请上传1M以下人脸清晰的照片", message=msg)
  38. xml = reply.render()
  39. resp.body = (xml)
  40. resp.status = falcon.HTTP_200
  41. app = falcon.API()
  42. connect = Connect()
  43. app.add_route("/connect", connect)

至此我们的工作就做完了,我们的公众号可以进行颜值检测了。本来我打算用在自己公众号上的,但是还存在下面几个问题,所以没有使用。

微信的机制,我们的程序必须在5s内给出响应。不然就会报"公众号提供的服务出现故障"。然而处理图片有时会比较慢,经常会超过5s。所以正确的处理方式应该是拿到用户的请求后立即返回一个空字符串表示我们收到了,之后多带带创建一个线程去处理图片,当图片处理完后通过客服接口发送给用户。可惜的是未认证的公众号没有客服接口,所以没办法,超过5s就会报错。

无法自定义菜单,一旦启用了自定义开发,菜单也需要自定义配置,但是未认证的公众号没有权限通过程序来配置菜单,只能在微信后台配置。

所以,我并没有在我的公众号上启用这个程序,但是如果有认证的公众号,可以尝试开发各种好玩的功能。

上篇:Python微信公众号开发—小白篇(一)

文章版权归作者所有,未经允许请勿转载,若此文章存在违规行为,您可以联系管理员删除。

转载请注明本文地址:https://www.ucloud.cn/yun/19755.html

相关文章

  • Python公众开发()—颜值检测

    摘要:上篇文章,我们把自己的程序接入了微信公众号,并且能把用户发送的文本及图片文件原样返回。微信的机制,我们的程序必须在内给出响应。上篇微信公众号开发小白篇一 上篇文章,我们把自己的程序接入了微信公众号,并且能把用户发送的文本及图片文件原样返回。今天我们把用户的图片通过腾讯的AI平台分析后再返回给用户。 为了防止我的文章被到处转载,贴一下我的公众号【智能制造社区】,欢迎大家关注。 githu...

    ?xiaoxiao, 评论0 收藏0
  • python机器/深度学习项目汇总

    摘要:作者微信公众号的皮卡丘欢迎大家搜索关注知乎机器学习美颜简单品读小说实现颜值预测预测比赛结果制作且版飞机大战实现简单的换脸术遗传算法拟合图像实现猫脸检测分析个人音乐收藏垃圾邮件识别深度学习强化学习玩破解游戏实现简单的机器翻译模型学写作实现文本 作者:Charles微信公众号:Charles的皮卡丘(欢迎大家搜索关注)知乎:https://zhuanlan.zhihu.com/p/... ...

    toddmark 评论0 收藏0
  • 首次公开,整理12年积累的博客收藏夹,零距离展示《收藏夹吃灰》系列博客

    摘要:时间永远都过得那么快,一晃从年注册,到现在已经过去了年那些被我藏在收藏夹吃灰的文章,已经太多了,是时候把他们整理一下了。那是因为收藏夹太乱,橡皮擦给设置私密了,不收拾不好看呀。 ...

    Harriet666 评论0 收藏0
  • Python 顶级开源项目 Top 10

    摘要:可以说是每个程序每天必逛的网站,这里集聚了全球最顶级的程序员。有非常多的大公司在上面开源自己的项目。今天整理了个上顶级的开源项目。这是推出的一款性能类型检查工具。它的主要目标是快速轻松地分发应用程序。 showImg(https://segmentfault.com/img/remote/1460000015909645); 阅读文本大概需要 4.2 分钟。 GitHub 可以说是每个...

    cangck_X 评论0 收藏0

发表评论

0条评论

最新活动
阅读需要支付1元查看
<