[jetson nx]使用face recognition

安装dlib

下载dlib
去官网下载

https://pypi.org/project/dlib/#history

或者pip下载

pip install dlib==19.22.0

face recognition

可以从我的码云克隆

git clone https://gitee.com/pro-kie/face_recognition.git
cd face_recognition
python3 setup.py build
python3 setup.py install

或者pip下载

pip install git+https://github.com/ageitgey/face_recognition_models

安装jetcam

git clone https://github.com/NVIDIA-AI-IOT/jetcam
cd jetcam
sudo python3 setup.py install

运行以下代码

# 人脸检测
from jetcam.csi_camera import CSICamera
import cv2 as cv
import numpy as np
from matplotlib import pyplot as plt
import face_recognition

camera0 = CSICamera(capture_device=0, width=512, height=512)


while 1:
    img = camera0.read()

    kk = cv.waitKey(1)
    if kk == ord('q'):  # 按下 q 键，退出
     break
    # trbl
    loc = face_recognition.face_locations(img=img,number_of_times_to_upsample=1,model='cnn')
    print(loc)
    if len(loc)==1:
        print(len(loc))
        loc = loc[0]
        print(loc)
        img = cv.rectangle(img,(loc[3],loc[0]),(loc[1],loc[2]),color=(0,0,255,20))
    cv.imshow('img',img)