短視訊直播原始碼，自動對上傳的圖片進行識別

短視訊直播原始碼，自動對上傳的圖片進行識別實現的相關程式碼

import face_recognition
import os
 
# 獲取圖片名列表
# 只讀取檔名以'.png'或'.jpg'或'.jpeg'結尾的圖片
unknow_people_list = [i for i in os.listdir('unknow_people') if (i.endswith('.jpg')) or (i.endswith('.png')) or (i.endswith('.jpeg'))]
know_people_list = [i for i in os.listdir('know_people') if (i.endswith('.jpg')) or (i.endswith('.png')) or (i.endswith('.jpeg'))]
 
 
def face_select():
    # 定義flag
    flag = 0
    for unknow_people in unknow_people_list:
        # 讀取待識別圖片
        unknow = face_recognition.load_image_file('unknow_people/' + unknow_people)
        # 待識別圖片轉化為特徵向量
        unknow_encode = face_recognition.face_encodings(unknow)[0]
        for know_people in know_people_list:
            # 讀取計算機已經認識的圖片
            know = face_recognition.load_image_file('know_people/'+know_people)
            # 圖片轉化為特徵向量
            know_encode = face_recognition.face_encodings(know)[0]
            # 兩張圖片進行比較
            # tolerance刻畫了比較的難易程度，值越小越難
            res = face_recognition.compare_faces([know_encode],unknow_encode,tolerance=0.5)
            if res[0]:
                flag = 1
                break
            else:
                flag = 0
        if flag == 1:
            print(f'匹配{unknow_people.split(".")[0]}')
        else:
            print(f'未匹配{unknow_people.split(".")[0]}')
 
 
if __name__ == '__main__':
    face_select()

以上就是 短視訊直播原始碼，自動對上傳的圖片進行識別實現的相關程式碼，更多內容歡迎關注之後的文章