野火鲁班猫(arrch64架构debian)从零实现用MobileFaceNet算法进行实时人脸识别(三)用yolov5-face算法实现人脸检测

环境直接使用第一篇中安装好的环境即可

先clone yolov5-face项目

复制代码
git clone https://github.com/deepcam-cn/yolov5-face.git

并下载预训练权重文件yolov5n-face.pt

网盘链接: https://pan.baidu.com/s/1xsYns6cyB84aPDgXB7sNDQ 提取码: lw9j

(野火官方提供)

再找一张带人脸的图片test.jpg保存在yolov5-face项目文件夹下

使用以下命令开始运行人脸检测任务

bash 复制代码
python detect_face.py --weights ./yolov5n-face.pt --img-size 640 --source ./test.jpg --view-img

这个时候,只要报错:

bash 复制代码
ModuleNotFoundError: No module named 'xxx'

就使用下面的命令安装一下相关的库就好

bash 复制代码
pip install xxx

再运行上面的人脸检测任务,我这又出现错误:

bash 复制代码
RuntimeError: Couldn't load custom C++ ops. This can happen if your PyTorch and torchvision versions are incompatible, or if you had errors while compiling torchvision from source. For further information on the compatible versions, check https://github.com/pytorch/vision#installation for the compatibility matrix. Please check your PyTorch version with torch.__version__ and your torchvision version with torchvision.__version__ and verify if they are compatible, and if not please reinstall torchvision so that it matches your PyTorch install.

这个错误是当前环境中安装的 PyTorchtorchvision二者版本不匹配,导致无法加载自定义 C++ 扩展导致的

使用以下代码查看PyTorchtorchvision当前的版本

bash 复制代码
python - <<EOF
import torch, torchvision
print("PyTorch:", torch.__version__)
print("torchvision:", torchvision.__version__)
EOF

我这显示

bash 复制代码
PyTorch: 2.3.0
torchvision: 0.15.2a0

解决办法:先卸载当前安装的pytorch和torchvision,再重新安装指定相匹配版本的pytorch和torchvision

bash 复制代码
# 卸载
pip uninstall torch torchvision -y

# 重新安装指定版本
pip install torch==2.3.0 torchvision==0.18.0 --extra-index-url https://download.pytorch.org/whl/cu121

再次执行人脸检测任务,虽然成功检测出来人脸,但是显示图片时又出现错误:

bash 复制代码
libGL error: failed to create dri screen
libGL error: failed to load driver: rockchip
libGL error: failed to create dri screen
libGL error: failed to load driver: rockchip
qt.qpa.xcb: QXcbConnection: XCB error: 148 (Unknown), sequence: 192, resource id: 0, major code: 140 (Unknown), minor code: 20

这个问题是没能正常加载Rockchip GPU 的驱动程序导致openGL加载失败,安装驱动解决问题:

bash 复制代码
sudo apt-get install mesa-utils libdrm-dev libegl-dev libgles2-mesa-dev

再次执行检测任务

bash 复制代码
python detect_face.py --weights ./yolov5n-face.pt --img-size 640 --source ./test.jpg --view-img

显示检测成功,图片也能正常显示

bash 复制代码
loading images ./test.jpg
image 1/1 /home/cat/yolov5/yolov5-face/test.jpg: 10 faces

使用rtsp地址代替图片,即可进行监控画面实时人脸检测

bash 复制代码
python detect_face.py --weights ./yolov5n-face.pt --img-size 640 --source rtsp://192.168.31.217:554/ch01.264 --view-img
相关推荐
魂尾ac几秒前
因为想开发新项目了~~要给老Python项目整个虚拟环境
开发语言·python
虎哥AI工具6 分钟前
Code Buddy 和 Cursor 到底谁更强?别再猜了!我花一下午,把代码、Bug、效率全给你们测明白了!
人工智能
LYue~7 分钟前
7月31号打卡
python
用户461757524706516 分钟前
小白理解transformer里的前馈神经网络
人工智能
柠檬味拥抱16 分钟前
AI Agent多模态融合策略研究与实证应用
人工智能
yj155817 分钟前
装修风格怎么选
python
泰Tai19 分钟前
【YOLOv8学习】WSL2中安装标注软件X-AnyLabeling
python·图像识别
笔触狂放19 分钟前
【机器学习】第九章 综合实战
人工智能·机器学习
玄明Hanko19 分钟前
从需求、开发、测试到运维,程序员效率飙升的秘密曝光
人工智能·ai编程
玄明Hanko20 分钟前
百度开源 ERNIE 4.5,将给国内大模型生态带来哪些影响
人工智能·文心一言·ai编程