facelib

from facelib import facerec
import cv2
import time


face_detector = facerec.SSDFaceDetector(tpu=True)
landmark_detector = facerec.LandmarkDetector(name='mobilenet_int8_tpu')
feature_extractor = facerec.FeatureExtractor(name='mobilenet_int8_tpu')

pipeline = facerec.Pipeline(face_detector, landmark_detector, feature_extractor)

path_img = '/home/pi/Desktop/images.jpeg'
img = cv2.imread(path_img)
#img = img[...,::-1] # cv2 returns bgr format but every method inside this package takes rgb format

inicio = time.time()
bboxes, landmarks, features = pipeline.predict(img)
print(time.time()-inicio)