利用D415读取 需要标记的人脸face_recognition的距离 Python + wind10

1、下载安装face_recognition

face_recognition 链接

2、文件结构如下

3、上代码

read_data.py

#-*-coding:utf8-*-
import os
#import cv2
#import numpy as np
import face_recognition
from read_img import endwith
import scipy.misc
 
def read_file(path):
    label_list = []
    dir_counter = 0

   
    img_encoding = [[] for i in range(5)]
    for child_dir in os.listdir(path):
         child_path = os.path.join(path, child_dir)

         for dir_image in  os.listdir(child_path):

             if endwith(dir_image,'jpg'):
                img = scipy.misc.imread(os.path.join(child_path, dir_image))
#                resized_img = cv2.resize(img, (IMG_SIZE, IMG_SIZE))
#                recolored_img = cv2.cvtColor(resized_img,cv2.COLOR_BGR2GRAY)
                img_encoding[dir_counter].append(face_recognition.face_encodings(img)[0])
#                img_encoding[dir_counter].append(face_recognition.face_encodings(img)[1])
#                img_encoding[dir_counter].append(face_recognition.face_encodings(img)[2])
                label_list.append(dir_counter)
         dir_counter += 1


    return img_encoding,label_list,dir_counter

 
def read_name_list(path):
    name_list = []
    for child_dir in os.listdir(path):
        name_list.append(child_dir)
    return name_list


 
if __name__ == '__main__':
    img_list,label_lsit,counter = read_file('./dataset')
    tt  = read_name_list('./dataset')
    print (img_list)
    print( tt)

read_img.py

import os
import cv2
 
def readAllImg(path,*suffix):
    try:
        s = os.listdir(path)
        resultArray = []
        fileName = os.path.b