HuggingFace:基于YOLOv8的人脸检测模型

个人操作经验总结

1、YOLO的环境配置

github

不论base环境版本如何,建议在conda的虚拟环境中安装

1.1、创建虚拟环境

python 复制代码
conda create -n yolov8-face python=3.9
复制代码
·conda create :创建conda虚拟环境,
·-n           :给虚拟环境命名的参数
·yolov8-face  :新建的虚拟环境叫做 yolov8-face
python=3.9    :指定安装的python版本

安装完成后进入yolov8-face虚拟环境

进入新建的虚拟环境

python 复制代码
conda activate yolov8-face

查看虚拟环境列表

python 复制代码
conda info --envs

1.2、安装torch【参考链接:最简单的一文安装Pytorch+CUDA

CPU版本torch安装

python 复制代码
pip install torch torchvision torchaudio  -i https://mirrors.tuna.tsinghua.edu.cn/pypi/web/simple

GPU版本(需要配置CUDA)

在配置好CUDA版本为12.4,安装torch

python 复制代码
pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu124

下载完torch之后,会提示一个numpy安装错误,此时需要重新下载一下numpy

python 复制代码
pip install numpy

测试GPU版本torch是否可用

python 复制代码
import torch 
torch.cuda.is_available()

返回True表示安装成功,如果返回False,那就下去沉淀沉淀,反思反思CUDA安装问题。

在代码测试中,发现GPT版本的Pytorch安装后,torchvision出现cpu版本问题。需要卸载重新安装

python 复制代码
pip uninstall torchvision

卸载后重新安装

python 复制代码
pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu124

1.3、Ultralytics YOLOv8安装

Ultralytics YOLOv8 是由 Ultralytics 开发的一个前沿的 SOTA 模型。它在以前成功的 YOLO 版本基础上,引入了新的功能和改进,进一步提升了其性能和灵活性。YOLOv8 基于快速、准确和易于使用的设计理念,使其成为广泛的目标检测、图像分割和图像分类任务的绝佳选择。

Ultralytics库包含YOLOv8的完整实现和所需的依赖。

python 复制代码
pip install ultralytics

2、模型下载

权重来自HuggingFace,有能力可以参考一下:arnabdhar/YOLOv8-Face-Detection

整体代码运行如下:

运行结果在result文件下

相关推荐
过往入尘土1 天前
YOLOv5:实时目标检测的现代化实践与深度解析
人工智能·yolo·目标检测
weixin_497845541 天前
通过ML.Net调用yolov5的Onnx模型
yolo·.net
FL16238631291 天前
基于YOLOv12和MediaPipe的人体跌倒检测系统
yolo
B站计算机毕业设计之家1 天前
计算机视觉:pyqt5+yoloV5目标检测平台 python实战 torch 目标识别 大数据项目 目标跟踪(建议收藏)✅
深度学习·qt·opencv·yolo·目标检测·计算机视觉·1024程序员节
知来者逆1 天前
计算机视觉——从环境配置到跨线计数的完整实现基于 YOLOv12 与质心追踪器的实时人员监控系统
人工智能·yolo·目标检测·计算机视觉·1024程序员节·目标追踪·yolov12
weixin_457340211 天前
VOC XML 旋转框 转换为 YOLO OBB 格式
xml·yolo·目标检测·视觉检测
lxmyzzs1 天前
【图像算法 - 30】基于深度学习的PCB板缺陷检测系统: YOLOv11 + UI界面 + 数据集实现
人工智能·深度学习·算法·yolo·缺陷检测
MADAO_luv1 天前
YOLOV11分类动物图像
深度学习·yolo·机器学习
AI浩2 天前
自监督 YOLO:利用对比学习实现标签高效的目标检测
学习·yolo·目标检测
newxtc2 天前
【江苏政务服务网-注册_登录安全分析报告】
人工智能·安全·yolo·政务·1024程序员节·安全爆破