欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页  >  IT编程

Python调用百度AI实现颜值评分功能

程序员文章站 2022-04-10 10:01:36
目录一、调用百度接口进行人脸属性识别二、根据年龄和性别对颜值进行评价三、批量识别人脸属性四、自定义窗口语音播报颜值得分五、明星颜值评价一、调用百度接口进行人脸属性识别安装好baidu-aip模块,获取...

一、调用百度接口进行人脸属性识别

安装好baidu-aip模块,获取了百度ai接口密钥后,即可调用百度接口进行人脸属性识别了。首先以杨紫的图片为例进行年龄、性别、颜值的识别。

Python调用百度AI实现颜值评分功能

具体python代码如下:‍

import os
import base64
from aip import aipface


os.chdir(r'f:\公众号\28.人脸识别')
#设置图片存放的路径

picturename = '1_yz.jpg'
def get_picture_content(picturename):
    with open(picturename, 'rb') as fp:
        content = base64.b64encode(fp.read())
        return content.decode('utf-8')
#定义读取图片的函数


app_id = 'xxx'
api_key = 'xxxxxxxx'
secret_key = 'xxxxxxxxxxxx'
#百度账号和密钥


options = {}
imagetype = 'base64'
options["face_field"] = "age, gender, beauty"
aipface = aipface(app_id, api_key, secret_key)
result = aipface.detect(get_picture_content(picturename), imagetype, options)
print(result)

注: 其中os.chdir中的内容应该替换成你存储图片的地址,app_id、api_key、secret_key应该替换成你获取的百度密钥。

得到结果如下:

Python调用百度AI实现颜值评分功能

其中,age对应的值为年龄,gender对应的值为性别,beauty对应的值为颜值分。为进一步规范输出成标准格式,通过如下代码进行规范化:

import pandas as pd

face_character = pd.dataframe({"age":[result['result']['face_list'][0]['age']], "gender":[result['result']['face_list'][0]['gender']['type']], "beauty":[result['result']['face_list'][0]['beauty']] })

得到结果如下:

Python调用百度AI实现颜值评分功能

可以发现杨紫的颜值得分还是很高的图片,后续文章也会对其它一些明星进行测试。

二、根据年龄和性别对颜值进行评价

根据性别和颜值分,对识别结果进行评价,具体代码如下:

if face_character['beauty'][0]>80:
    if face_character['gender'][0] == 'female':
        print("小姐姐颜值爆表")
    else:
        print("小哥哥颜值爆表") 
elif face_character['beauty'][0]>70:
    if face_character['gender'][0] == 'female':
        print("小姐姐天生美颜")
    else:
        print("小哥哥天生美颜") 
elif face_character['beauty'][0]>60:
    if face_character['gender'][0] == 'female':
        print("小姐姐颜值尚可")
    else:
        print("小哥哥颜值尚可") 
elif face_character['beauty'][0]>40:
    if face_character['gender'][0] == 'female':
        print("小姐姐先天不够,后天来凑")
    else:
        print("小哥哥先天不够,后天来凑") 
else:
    if face_character['gender'][0] == 'female':
        print("小姐姐洗洗睡吧")
    else:
        print("小哥哥洗洗睡吧")

得到结果:

Python调用百度AI实现颜值评分功能

三、批量识别人脸属性

假设文件夹中有数张人脸图片,想批量对这批图片进行颜值打分,可执行如下代码:

import os
import re
import time
import base64
from aip import aipface


os.chdir(r'f:\公众号\28.人脸识别')
#设置图片存放的路径

picturename = '6_brzp.jpg'
def get_picture_content(picturename):
    with open(picturename, 'rb') as fp:
        content = base64.b64encode(fp.read())
        return content.decode()
#定义读取图片的函数

app_id = 'xxx'
api_key = 'xxxxxxxx'
secret_key = 'xxxxxxxxxxxx'
#百度账号和密钥


options = {}
imagetype = 'base64'
options["face_field"] = "beauty"
options["max_face_num"] = 2
aipface = aipface(app_id, api_key, secret_key)

root_path = r'f:\公众号\28.人脸识别'
list = os.listdir(root_path) 
# 列出文件夹下所有的目录与文件

for i in range(0, len(list)):
    time.sleep(1)  
    path = os.path.join(root_path, list[i])
    if os.path.isfile(path):
        result = aipface.detect(get_picture_content(path), imagetype, options)
        print(list[i] + ' 评分为:' + str(result['result']['face_list'][0]['beauty']))
#显示出所有图片的颜值得分

得到结果:

Python调用百度AI实现颜值评分功能

可以发现杨紫的那张照片颜值分很高,有八十几分。为了更好的用户体验,我们设置一个窗口,显示评价照片并语音播报颜值得分。

四、自定义窗口语音播报颜值得分

为了更清晰地评价人的颜值,把图片和结果用tkinter库和语音播放库包装起来,效果会更好。具体代码如下:

import os
import re
import time
import base64
import pandas as pd
import tkinter as tk
from aip import aipface
from tkinter import filedialog

root = tk.tk()
root.withdraw()
folderpath = filedialog.askdirectory(title = '请选择图片存放的文件夹')  
picturepath = filedialog.askopenfilename(title = '请选择要进行颜值评价的图片')
root.destroy()
os.chdir(folderpath)
#设置图片存放的路径

def get_picture_content(picturepath):
    with open(picturepath, 'rb') as fp:
        content = base64.b64encode(fp.read())
        return content.decode()
#定义读取图片的函数

app_id = 'xxx'
api_key = 'xxxxxxxx'
secret_key = 'xxxxxxxxxxxx'
#百度账号和密钥


options = {}
options["max_face_num"] = 2
options["face_field"] = "gender"
aipface = aipface(app_id, api_key, secret_key)
imagetype = 'base64'
result = aipface.detect(get_picture_content(picturepath), imagetype, options)
gender = result['result']['face_list'][0]['gender']['type']
options["face_field"] = "age"
time.sleep(1)
result = aipface.detect(get_picture_content(picturepath), imagetype, options)
age = result['result']['face_list'][0]['age']
options["face_field"] = "beauty"
time.sleep(2)
result = aipface.detect(get_picture_content(picturepath), imagetype, options)
beauty = result['result']['face_list'][0]['beauty']
#获取年龄、性别、颜值信息

face_character = pd.dataframe({"age":age, "gender":gender, "beauty":beauty},index = ['value'])
if face_character['beauty'][0]>80:
    if face_character['gender'][0] == 'female':
        str_list = "小姐姐芳龄:"+ str(age) + ",颜值爆表"+ ',最终颜值得分为:'+str(beauty)
    else:
        str_list = "小哥哥贵庚:"+ str(age) + ",颜值爆表"+ ',最终颜值得分为:'+str(beauty)
elif face_character['beauty'][0]>70:
    if face_character['gender'][0] == 'female':
        str_list = "小姐姐芳龄:"+ str(age) + ",天生美颜"+ ',最终颜值得分为:'+str(beauty)
    else:
        str_list = "小哥哥贵庚:"+ str(age) + ",天生美颜"+ ',最终颜值得分为:'+str(beauty)
elif face_character['beauty'][0]>50:
    if face_character['gender'][0] == 'female':
        str_list = "小姐姐芳龄:"+ str(age) + ",颜值尚可"+ ',最终颜值得分为:'+str(beauty)
    else:
        str_list = "小哥哥贵庚:"+ str(age) + ",颜值尚可"+ ',最终颜值得分为:'+str(beauty)
elif face_character['beauty'][0]>30:
    if face_character['gender'][0] == 'female':
        str_list = "小姐姐芳龄:"+ str(age) + ",先天不够,后天来凑"+ ',最终颜值得分为:'+str(beauty)
    else:
        str_list = "小哥哥贵庚:"+ str(age) + ",先天不够,后天来凑" + ',最终颜值得分为:'+str(beauty)
else:
    if face_character['gender'][0] == 'female':
        str_list = "小姐姐芳龄:"+ str(age) + ",早点洗洗睡吧"+ ',最终颜值得分为:'+str(beauty)
    else:
        str_list = "小哥哥贵庚:"+ str(age) + ",早点洗洗睡吧"+ ',最终颜值得分为:'+str(beauty)
#颜值定义

from tkinter import *
from pil import image, imagetk
from win32com.client import dispatch
#导入包


speaker = dispatch("sapi.spvoice")
def roll_call():
    speaker.speak(str_list)
#控制播放语音


os.chdir(folderpath)
#设置文件路径

root = tk()
root.title("颜值测试小程序")
root.iconbitmap("pikaqiu2.ico")
#设置窗口

image = image.open(picturepath)
#加载图片

root.geometry("400x300")
#根据图片大小设置窗口大小

img_pic = imagetk.photoimage(image)
label = label(root, image=img_pic)
label.pack()

b2 = tk.button(root, bg='lightyellow', text='颜值评价', font=("kaiti", 8), width=8, height=2, command=roll_call)
b2.place(x=0, y=0)

root.mainloop()

五、明星颜值评价

最后我们来看下明星的颜值评分吧。

1 刘亦菲

Python调用百度AI实现颜值评分功能

颜值评价结果:

'小姐姐芳龄:21,颜值尚可,最终颜值得分为:64.19'

对于这个结果,我也比较吃惊,所以颜值得分大家当成娱乐就好图片。

2 贾玲

Python调用百度AI实现颜值评分功能

颜值评价结果:

'小姐姐芳龄:37,先天不够,后天来凑,最终颜值得分为:30.67'

颜值得分仅供参考,请勿当真。我的有些照片测出来也是三十多分,可能跟背景、光线、表情都有一定的关系,大家当成娱乐就好。

3 肖战

Python调用百度AI实现颜值评分功能

颜值评价结果:

'小哥哥贵庚:23,颜值尚可,最终颜值得分为:63.9'

颜值得分仅供参考,请勿当真图片,认为我的代码有问题的可以自行下载图片和代码进行尝试。

4 宋小宝

Python调用百度AI实现颜值评分功能

颜值评价结果:

'小哥哥贵庚:40,先天不够,后天来凑,最终颜值得分为:45.18'

颜值得分仅供参考,请勿当真。从上面的颜值得分可以发现,普遍颜值得分较低,八十多分的已经算高分了,也算是为杨紫的颜值正名了吧。至此,调用百度接口颜值识别已讲解完毕,感兴趣的朋友自己实现一遍吧。 

以上就是python调用百度ai实现颜值评分功能的详细内容,更多关于python 颜值评分的资料请关注其它相关文章!