yolov8环境搭建与应用

YOLOv8(You Only Look Once version 8)是一个深度学习框架,用于实现实时对象检测。它是 YOLO 系列的最新迭代,旨在提供更高的准确性和速度。YOLOv8 继承了前代模型的优点,并在此基础上进行了多项改进,包括更复杂的网络架构、更优化的训练流程和更强大的特征提取能力。这些改进共同作用,使得 YOLOv8 在多个标准数据集上达到了前所未有的检测性能。

YOLOv8 的特点:

实时性能: YOLOv8 继续保持 YOLO 系列的实时检测特性,即使在较低的硬件配置上也能达到很高的帧率(FPS)。

高准确度: 通过更深更复杂的网络结构和改进的训练技巧,YOLOv8 在保持高速度的同时,也大幅提高了检测的准确度。

多尺度预测: YOLOv8 引入了改进的多尺度预测技术,可以更好地检测不同大小的对象。

自适应锚框: 新版在自适应调整锚框方面做了优化,可以更准确地预测对象的位置和大小。

一、环境安装

1.1 miniconda安装(可选,可用anaconda也可直接用pip)

1.1.1miniconda安装

  • 如果C盘有空间,最好安装在C盘,且安装目录不要有中文
  • 勾选将其添加到Path
1.1.2 conda环境创建
#创建命令。明确指定版本,否则可能会因为版本过高导致有包安装不上
conda create -n yolov8 python=3.9

#激活环境
conda activate yolov8

#查看环境列表
conda env list
1.1.3 pip配置国内源(阿里源)
pip config set global.index-url https://mirrors.aliyun.com/pypi/simple/
pip config set global.trusted-host mirrors.aliyun.com
#查看pip镜像源
pip config list

1.2 pytorch安装

1.2.1 pytorch安装
  • 官网地址:链接
  • 在一个单独的环境中,能使用pip就尽量使用pip,实在有问题的情况,例如没有合适的编译好的系统版本的安装包,再使用conda进行安装,不要来回混淆
  • 根据你自己的情况选择版本与安装方式

1.2.2 CUDA是否需要安装
        如果只需要训练、简单推理,则无需单独安装CUDA,直接安装pytorch,如果有部署需求,例如导出TensorRT模型,则需要进行CUDA安装
1.2.2.1 首先查看自身电脑的是否有NVIDIA硬件环境
在电脑桌面右键,打开NVIDIA的控制面板

1.2.2.2 下载对应cuda版本的驱动

下载地址:链接

选择自己对应的版本下载

1.2.2.3 下载安装cuDNN
http官网地址 https://developer.nvidia.com/cudnn-downloads 点我跳转
下载cuDNN是需要登录英伟达开发者账户的,注册一个并填写问卷就行了,很简单。
注意:必须选择和你安装的CUDA匹配的版本,我这里选中12.0.1版本的。

  • 将cuDNN解压到D盘
  • 将三个文件夹拷贝到到cuda的安装目录下。默认的安装路径为C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v12.0
  • 添加环境变量
  • 1 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v12.0\bin
    2 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v12.0\include
    3 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v12.0\lib
    4 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v12.0\libnvvp
    

    -验证: win+R cmd进入安装目录下,再进入到 extras\demo_suite下,执行.\bandwidthTest.exe和.\deviceQuery.exe,得到下图。

1.2.3Pytorch安装注意事项
  • 16XX的显卡,安装cu102版本,否则可能训练出现问题
  • 30xx、40xx显卡,需要安装cu111以上版本,否则无法运行
  • 这里我的显卡事3050的,使用的安装命令是
pip install torch==1.13.1+cu116 torchvision==0.14.1+cu116 torchaudio==0.13.1 --extra-index-url https://download.pytorch.org/whl/cu116

1.3 ultralytics(YOLOv8)安装

  • 直接使用源码(不是很推荐,无法使用命令行工具) github地址
  • pip直接安装(官方推荐,个人不是很推荐)
  • pip源码安装(个人推荐)
1.下载github源码
2.解压源码
3.进入解压的ultralytics-main文件夹
4.执行pip install -e .  (-e . 参数必须要有,否则后续修改代码无效)

安装好之后,使用 pip list 查看

后边的后缀表示这个包的源码在哪个位置

可能会出现的问题:ERROR: Cannot uninstall ‘TBB’. It is a distutils installed project and thus we cannot accurately determine which files belong to it which would lead to only a partial uninstall.

  1. 执行pip - V (大写V),会出现一个目录
  2. 进入目录 Lib\site-packages 删除文件夹:“llvmlite”
  3. 然后执行 conda uninstall TBB 中间需要输入一次Y
  4. 然后再重新安即可装
1.3.1 测试

先进入ultralytics-main目录,然后激活环境

conda activate yolov8

再执行命令

#这个命令需要下载模型,网络不稳定可能需要科学上网
yolo predict model=yolov8n.pt source=ultralytics/assets/bus.jpg

1.4 正确使用windows中的终端

因为此前已经创建过虚拟环境了,激活环境

conda activate yolov8

二、模型预测

环境验证(可选)

import torch

print(torch.__version__)#输出PyTorch版本
print(torch.cuda.is_available())#这个理论输出True
print(torch.version.cuda)#这个应该输出CUDA版本
print(torch.backends.cudnn.version())#输出CUDNN版本
print(torch.cuda.device_count())#这个应该输出GPU数量

2.1 模型预测的基本使用

from ultralytics import YOLO

yolo = YOLO("./yolov8n.pt" ,task = "detect")
# result=yolo(source="./ultralytics/assets/bus.jpg")#对图片进行检测/这里也可以是视频
# result=yolo(source="screen")#对屏幕进行检测
# result=yolo(source=0)#对摄像头进行检测

# result=yolo(source="./ultralytics/assets/bus.jpg",save=True)#save=True 对
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值