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

crownyouyou2024-11-19 2:07

个人操作经验总结

1、YOLO的环境配置

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

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

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

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

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

python 复制代码

pip install ultralytics

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

整体代码运行如下：

运行结果在result文件下