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文件下

相关推荐
Hcoco_me9 小时前
YOLO入门教程(番外):目标检测的一阶段学习方法
yolo·目标检测·学习方法
七芒星202312 小时前
多目标识别YOLO :YOLOV3 原理
图像处理·人工智能·yolo·计算机视觉·目标跟踪·分类·聚类
深瞳智检15 小时前
YOLO算法原理详解系列 第007期-YOLOv7 算法原理详解
人工智能·算法·yolo·目标检测·计算机视觉·目标跟踪
羊羊小栈16 小时前
基于「YOLO目标检测 + 多模态AI分析」的光伏板缺陷检测分析系统(vue+flask+模型训练+AI算法)
vue.js·人工智能·yolo·目标检测·flask·毕业设计·大作业
B站计算机毕业设计之家1 天前
智慧交通项目:Python+PySide6 车辆检测系统 YOLOv8+OpenCV 自定义视频 自定义检测区域 (源码+文档)✅
大数据·python·opencv·yolo·智慧交通·交通·车流量
Mrs.Gril3 天前
目标检测:yolov7算法在RK3588上部署
算法·yolo·目标检测
格林威3 天前
工业视觉检测里的 “柔性” 是什么?
图像处理·人工智能·深度学习·yolo·计算机视觉·视觉检测
霍夫曼vx_helloworld73523 天前
yolov8模型在指针式表盘读数中的应用【代码+数据集+python环境+GUI系统】
大数据·python·yolo
飞翔的佩奇4 天前
【完整源码+数据集+部署教程】 白血球图像分割系统: yolov8-seg-repvit
python·yolo·计算机视觉·数据集·yolo11·白血球图像分割系统·、yolov8
Hcoco_me4 天前
YOLO入门教程(番外):卷积神经网络—图像卷积
深度学习·yolo·cnn