减少保存编码的时间人脸识别

时间:2018-06-24 06:28:54

标签: python pickle

我有一个数据集,我正在将图像的编码结果保存下来,以便在酱菜对象中进行最爱识别。 我想在数据库中添加新图像或删除图像,当我这样做时,数据库中存在的先前图像将存储在dataset_faces.dat中,并且仅对新图像完成encode_faces.py。 我想减少将编码保存在encoding.pickle中的时间。 否则,即使添加新图像也应花费大量时间。

encode_faces.py

import face_recognition
import numpy as np
import os
import pickle
known_person = []
known_image= []
known_face_encoding=[]

for file in os.listdir("Imagefolder"):

        #Extracting person name from the image filename eg:Abhilash.jpg
        known_person.append(str(file).replace(".jpg", ""))
        file=os.path.join("Imagefolder", file)
        known_image = face_recognition.load_image_file(file)
        known_face_encoding.append(face_recognition.face_encodings(known_image)[0])

        with open('dataset_faces.dat', 'wb') as f:
             pickle.dump(known_face_encoding, f,pickle.HIGHEST_PROTOCOL)

        with open('dataset_fac.dat', 'wb') as d:
             pickle.dump(known_person, d)



print(known_face_encoding)
print(known_person)

0 个答案:

没有答案