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

embedded/2024/11/19 3:07:06/

个人操作经验总结

YOLO_1">1、YOLO的环境配置

github

在这里插入图片描述
不论base环境版本如何,建议在conda的虚拟环境中安装

1.1、创建虚拟环境

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

在这里插入图片描述

安装完成后进入yolov8-face虚拟环境
在这里插入图片描述
进入新建的虚拟环境

conda activate yolov8-face

查看虚拟环境列表

conda info --envs

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

CPU版本torch安装

pip install torch torchvision torchaudio  -i https://mirrors.tuna.tsinghua.edu.cn/pypi/web/simple

GPU版本(需要配置CUDA)
在配置好CUDA版本为12.4,安装torch

pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu124

在这里插入图片描述
下载完torch之后,会提示一个numpy安装错误,此时需要重新下载一下numpy

pip install numpy

测试GPU版本torch是否可用

import torch 
torch.cuda.is_available()

在这里插入图片描述
返回True表示安装成功,如果返回False,那就下去沉淀沉淀,反思反思CUDA安装问题。

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

pip uninstall torchvision

卸载后重新安装

pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu124

YOLOv8_69">1.3、Ultralytics YOLOv8安装

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

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

pip install ultralytics

在这里插入图片描述

2、模型下载

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

整体代码运行如下:在这里插入图片描述
运行结果在result文件下
在这里插入图片描述


http://www.ppmy.cn/embedded/138674.html

相关文章

基于BERT的情感分析

基于BERT的情感分析 1. 项目背景 情感分析(Sentiment Analysis)是自然语言处理的重要应用之一,用于判断文本的情感倾向,如正面、负面或中性。随着深度学习的发展,预训练语言模型如BERT在各种自然语言处理任务中取得了…

几何合理的分片段感知的3D分子生成 FragGen - 评测

FragGen 来源于 2024 年 3 月 25 日 预印本的文章,文章题目是 Deep Geometry Handling and Fragment-wise Molecular 3D Graph Generation, 作者是 Odin Zhang,侯廷军,浙江大学药学院。FragGen 是一个基于分子片段的 3D 分子生成模…

Android 中的 Zygote 和 Copy-on-Write 机制详解

在 Android 系统中,Zygote 是一个关键的进程,几乎所有的应用进程都是通过它 fork(派生)出来的。通过 Zygote 启动新进程的方式带来了显著的性能优势,这得益于 fork 操作和 Linux 中的 Copy-on-Write(COW&am…

预处理(1)(手绘)

大家好,今天给大家分享一下编译器预处理阶段,那么我们来看看。 上面是一些预处理阶段的知识,那么明天给大家讲讲宏吧。 今天分享就到这里,谢谢大家!!

nginx配置负载均衡详解

在现代的 web 应用中,负载均衡是确保高可用性、可扩展性和稳定性的关键技术之一。Nginx 是一个非常流行的反向代理服务器和负载均衡器,它支持多种负载均衡策略,能够帮助将客户端的请求分发到多个后端服务器,以提高系统的整体性能和…

STM32单片机设计防儿童人员误锁/滞留车内警报系统

目录 目录 前言 一、本设计主要实现哪些很“开门”功能? 二、电路设计原理图 1.电路图采用Altium Designer进行设计: 2.实物展示图片 三、程序源代码设计 四、获取资料内容 前言 近年来在车辆逐渐普及的情况下,由于家长的疏忽,将…

STL关联式容器介绍

在前文中介绍了STL的序列式容器; STL序列式容器之vector-CSDN博客 STL序列式容器之list-CSDN博客 STL序列式容器之deque-CSDN博客 STL序列式容器之stack-CSDN博客 STL序列式容器之queue-CSDN博客 STL序列式容器之heap(堆)-CSDN博客 ST…

HTML之列表学习记录

练习题&#xff1a; 图所示为一个问卷调查网页&#xff0c;请制作出来。要求&#xff1a;大标题用h1标签&#xff1b;小题目用h3标签&#xff1b;前两个问题使用有序列表&#xff1b;最后一个问题使用无序列表。 代码&#xff1a; <!DOCTYPE html> <html> <he…