如何使用python寫個顏值評分器

小編給大家分享一下如何使用python寫個顏值評分器，相信大部分人都還不怎么了解，因此分享這篇文章給大家參考一下，希望大家閱讀完這篇文章后大有收獲，下面讓我們一起去了解一下吧！

一、核心功能設(shè)計

總體來說，我們需要做的是獲取直播顏值區(qū)的主播小姐姐的正在直播的全部主播名稱和封面圖并保存下來，用百度AI提供的人臉識別接口，進行顏值評分排序，選出顏值最高的。

拆解需求，大致可以整理出核心功能如下：

獲取主播直播封面圖

• 打開直播顏值區(qū)模塊對頁面進行分析

• 發(fā)送網(wǎng)絡(luò)請求，解析數(shù)據(jù)

• 保存數(shù)據(jù)

主播顏值評分

• 百度人臉識別接口

• 遍歷主播照片，調(diào)用顏值檢測接口對主播顏值進行打分

• 對評分進行排序

二、實現(xiàn)步驟

1. 獲取主播名稱和照片

首先我們選擇的是某牙直播，進入首頁打開顏值區(qū)，按F12可以進入開發(fā)者模式。

import requests
# 1.找到數(shù)據(jù)所在url地址（系統(tǒng)分析網(wǎng)頁性質(zhì)）
url = "https://www.huya.com/g/2168"
headers = {
        'User-Agent': 'User-Agent: Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 '
                      '(KHTML, like Gecko) Chrome/87.0.4280.88 Safari/537.36'
    }

# 2. 發(fā)送網(wǎng)絡(luò)請求
response = requests.get(url=url, headers=headers)
html_data = response.text
print(html_data)

不難發(fā)現(xiàn)所有的小姐姐直播封面對應(yīng)的都是在li標(biāo)簽里面。我們只要解析獲取這些li標(biāo)簽數(shù)據(jù)就可以了。

接著我們需要拿到直播小姐姐的封面圖片，通過分析上面li標(biāo)簽里面的內(nèi)容，可以發(fā)現(xiàn)下面有個a標(biāo)簽，里面的img標(biāo)簽中的data-original不就是我們要的小姐姐圖片嘛！

接下來我們想要獲取主播小姐姐的名字怎么辦呢？點開li標(biāo)簽繼續(xù)分析，可以看到下面有個span標(biāo)簽，其中的i標(biāo)簽內(nèi)容就是小姐姐直播的名字。

網(wǎng)頁結(jié)構(gòu)我們上面已經(jīng)分析好了，那么我們就可以來動手爬取我們所需要的數(shù)據(jù)了。獲取到所有的數(shù)據(jù)資源之后，把圖片保存下來即可。文件的下載保存的方式比較多，我用的是通過 with open打開文件的方式 ，比較簡單。

# 3. 數(shù)據(jù)解析
selector = parsel.Selector(html_data)
lis = selector.xpath('//li[@class="game-live-item"]')  # 所有l(wèi)i標(biāo)簽

for li in lis:
    img_name = li.xpath('.//span[@class="avatar fl"]/i/text()').get()  # 主播名字
    img_url = li.xpath('.//a/img/@data-original').get()  # 主播圖片地址
    # print(img_name, img_url)

    # 請求圖片數(shù)據(jù)
    img_data = requests.get(url=img_url).content  # 圖片數(shù)據(jù)

    # 4. 數(shù)據(jù)保存
    # 準(zhǔn)備文件名
    file_name = img_name + '.jpg'
    with open('img\\' + file_name, mode='wb') as f:
        f.write(img_data)
        print('正在保存:', file_name)

這樣小姐姐的直播名稱和照片都可以保存下來了，效果如下：

2. 主播顏值評分

我們調(diào)用的是百度開放的人臉識別接口 – 百度AI開放平臺鏈接。

這里面我們可以創(chuàng)建一個人臉識別應(yīng)用，其中的API Key及Secret Key后面我們調(diào)用人臉識別檢測接口時會用到。

接下來我們可以看看官方提供的API幫助文檔，里面介紹的很詳細(xì)。包括如何調(diào)用請求URL數(shù)據(jù)格式，向API服務(wù)地址使用POST發(fā)送請求，必須在URL中帶上參數(shù)access_token，可通過后臺的API Key和Secret Key生成。這里面的API Key和Secret Key就是我們上面提到的。

那我們要的打分顏值分?jǐn)?shù)是哪個呢？提供返回結(jié)果參數(shù)，可以看到里面有個beauty就是我們要的顏值分?jǐn)?shù)。

這樣顏值檢測的接口流程基本就已經(jīng)清楚了，可以進行代碼實現(xiàn)了。

其中獲取token的時候，需要用到client_id 和 client_secret ，這兩個就是上面創(chuàng)建人臉識別應(yīng)用時提供的。

import base64
import requests
# import pprint


# 獲取token
def get_token():
    # client_id 為官網(wǎng)獲取的AK， client_secret 為官網(wǎng)獲取的SK
    host = 'https://aip.baidubce.com/oauth/2.0/token?grant_type=client_credentials&client_id=【官網(wǎng)獲取的AK】&client_secret=【官網(wǎng)獲取的SK】'
    response = requests.get(host)
    if response:
        # print(response.json())
        return response.json()['access_token']


# 顏值檢測接口
def face_input(file_path):
    with open(file_path, 'rb') as file:
        data = base64.b64encode(file.read())
    img = data.decode()
    request_url = "https://aip.baidubce.com/rest/2.0/face/v3/detect"

    params = "{\"image\":\"%s\",\"image_type\":\"BASE64\",\"face_field\":\"beauty\"}" % img
    access_token = get_token()
    request_url = request_url + "?access_token=" + access_token
    headers = {'content-type': 'application/json'}
    response = requests.post(request_url, data=params, headers=headers)
    if response:
        beauty = response.json()['result']['face_list'][0]['beauty']
        # pprint.pprint(response.json())
        return beauty

可以看到result字段里面的beauty就是代表對小姐姐的顏值評分。效果如下：

調(diào)用顏值檢測接口已經(jīng)寫好了，下面我們要遍歷之前保存的所有小姐姐直播照片，對每個進行顏值打分。

path = './img'
img_list = os.listdir(path)
# print(img_list)
score_dict ={}

for img in img_list:
    try:
        # 提取主播名字
        name = img.split('.')[0]
        # 構(gòu)建圖片路徑
        img_path = path + '//' + img
        # 調(diào)用顏值檢測接口
        face_score = face_input(img_path)
        # print(face_score)
        score_dict[name] = face_score
    except:
        print(f'正在檢測{name}| 檢測失敗')
    else:
        print(f'正在檢測{name}| \t\t 顏值打分為：{face_score}')

最后我們就只需要按照顏值分?jǐn)?shù)進行降序排列，就可以選出顏值最高的小姐姐啦~

sorted_score = sorted(score_dict.items(), key=lambda x: x[1], reverse=True)
# print(sorted_score)

for i, j in enumerate(sorted_score):
    print(f'小姐姐名字是：{sorted_score[i][0]} | 顏值名次是：第{i+1}名 | 顏值分?jǐn)?shù)是：{sorted_score[i][1]}')

通過顏值檢測，這樣就可以找到顏值最高的小姐姐了，顏值打分有90分以上。今天我們就到這里，明天繼續(xù)努力！不說了，趕緊看直播去~

以上是“如何使用python寫個顏值評分器”這篇文章的所有內(nèi)容，感謝各位的閱讀！相信大家都有了一定的了解，希望分享的內(nèi)容對大家有所幫助，如果還想學(xué)習(xí)更多知識，歡迎關(guān)注億速云行業(yè)資訊頻道！