当前位置: 首页 > article >正文

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

个人操作经验总结

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

1.3、Ultralytics YOLOv8安装

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

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

pip install ultralytics

在这里插入图片描述

2、模型下载

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

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


http://www.kler.cn/a/399382.html

相关文章:

  • 深度学习笔记14-卷积神经网络2
  • Docker:查看镜像里的文件
  • 哈希表学习分享
  • ReactPress与WordPress:两大开源发布平台的对比与选择
  • js导入导出
  • PaoluGPT——千里挑一
  • xpath表达式学习总结
  • redis:RDB和AOF机制
  • 【算法】回文数索引、回文子串输出、整数反转
  • JavaScript 中的多重继承与 ES6 中的继承
  • Linux(光速安装+centos镜像 图片+大白话)
  • Python 小高考篇(7)常用模板
  • Python 小高考篇(6)常见错误及排查
  • Softmax Temperature
  • HarmonyOS:使用常用组件构建页面
  • uniapp vue3的下拉刷新和上拉加载
  • java 读取 有时需要sc.nextLine();读取换行符 有时不需要sc.nextLine();读取换行符 详解
  • 药香代码:Spring Boot中药实验管理实践
  • 在Flutter中,禁止侧滑的方法
  • 基于微信小程序的在线学习平台+LW示例参考
  • qt移植到讯为rk3568,包含一些错误总结
  • 2024 - 超火的多模态深度学习公共数据纯生信5+思路分享
  • 卡牌对弈游戏策略-贪心算法
  • 基于Python的仓库管理系统设计与实现
  • Element-ui Select选择器自定义搜索方法
  • 游戏如何应对内存修改