YOLOv8初体验:检测、跟踪、模型部署
创始人
2024-05-31 03:56:53
0

安装

YOLOv8有两种安装方式,一种是直接用pip命令安装:

pip install ultralytics

另外一种是通过源码安装:

git clone https://github.com/ultralytics/ultralytics
cd ultralytics
pip install -e '.[dev]'

安装完成后就可以通过yolo命令在命令行进行使用了。

目标检测

使用YOLOv8进行目标检测,可以使用下面的命令:

yolo task=detect mode=predict model=yolov8n.pt source=ultralytics/assets/bus.jpg imgsz=640 show=True save=True

如果模型权重不存在,程序会自动从GitHub中下载。如果对命令行的参数不了解,可以参考官方文档的说明,也可以直接看ultralytics代码仓库中的ultralytics/yolo/cfg/default.yaml文件,里面有所有参数的说明和默认值。

上面的命令运行成功后,带检测结果的图片被保存到runs/detect/predict目录下,如下图所示:

跑视频的效果可以看下面这个视频:

用TensorRT部署的YOLOv8模型,来看看效果怎么样

目标跟踪

YOLOv8目前支持BoT-SORTByteTrack两种多目标跟踪算法,默认的目标跟踪算法为BoT-SORT,使用方式如下:

yolo track model=yolov8n.pt source=test.avi show=True save=True

如果要使用ByteTrack跟踪算法,可以添加命令行参数tracker=bytetrack.yaml

下面的视频是使用BoT-SORT算法的跟踪效果,效果还是不错的。

YOLOv8+BoT-SORT目标检测与跟踪

模型部署

如果要用TensorRT部署YOLOv8,需要先使用下面的命令将模型导出为onnx格式:

yolo export model=yolov8n.pt format=onnx opset=12 

YOLOv83个检测头一共有80x80+40x40+20x20=8400个输出单元格,每个单元格包含x,y,w,h4项再加80个类别的置信度总共84项内容,所以通过上面命令导出的onnx模型的输出维度为1x84x8400

这样的通道排列顺序有个问题,那就是后处理的时候会造成内存访问不连续。为了解决这个问题,我们可以修改一下代码,具体做法是把ultralytics/nn/modules.py文件中的421行做如下修改,交换一下张量y的通道顺序:

修改代码后需要执行前面的安装命令pip install -e '.[dev]'使代码生效。这样修改后再执行上面的模型导出命令,模型的输出维度变为1x8400x84

导出onnx模型后,就可以用TensorRT进行部署了。如何使用TensorRTC++接口部署ONNX模型可以参考我之前给【自动驾驶之心】公众号写的这篇文章:手把手教学!TensorRT部署实战:YOLOv5的ONNX模型部署。

YOLOv5相比,部署YOLOv8的不同之处就是做后处理解析模型输出结果的时候不需要去解析objectness这项内容了,其他都类似。

使用TensorRT框架在我的GeForce GTX 1650 Ti显卡上部署YOLOv8的结果如下:

模型输入尺寸模型精度耗时(ms)
yolov8n.onnx640x640FP327
yolov8s.onnx640x640FP3212
yolov8m.onnx640x640FP3229
yolov8l.onnx640x640FP3252
yolov8x.onnx640x640FP3283
yolov8n.onnx640x640FP164
yolov8s.onnx640x640FP167
yolov8m.onnx640x640FP1614
yolov8l.onnx640x640FP1625
yolov8x.onnx640x640FP1640

YOLOv5测试结果:

模型输入尺寸模型精度耗时(ms)
yolov5n.onnx640x640FP327
yolov5s.onnx640x640FP3210
yolov5m.onnx640x640FP3221
yolov5l.onnx640x640FP3241
yolov5x.onnx640x640FP3276
yolov5n.onnx640x640FP165
yolov5s.onnx640x640FP166
yolov5m.onnx640x640FP1611
yolov5l.onnx640x640FP1621
yolov5x.onnx640x640FP1638

从上面的测试结果来看,YOLOv8YOLOv5稍微慢一点点。

相关内容

热门资讯

手机装虚拟安卓系统教程,手机安... 你有没有想过,你的手机可以变成一个多才多艺的小戏精,既能装Windows,又能玩安卓?没错,这就是今...
安卓系统平板电脑屏幕暗,打造舒... 你是不是也遇到了这样的烦恼?你的安卓系统平板电脑屏幕突然暗了下来,亮度调节按钮也似乎失灵了。别急,让...
安卓高清系统推荐电脑版,打造极... 你有没有想过,手机上的安卓高清系统竟然也能搬到电脑上?没错,就是那种流畅、美观,还能让你在电脑上也能...
自己刷安卓盒子系统,功能、操作... 你有没有想过,家里的安卓盒子系统是不是该来个焕然一新的大变身呢?别急,让我带你一步步走进自己刷安卓盒...
安卓系统的空格怎么输入,探索安... 你是不是也和我一样,在使用安卓手机的时候,有时候会遇到一个让人头疼的问题——怎么输入空格呢?别急,今...
安卓系统如何删除僵尸粉 你是不是也遇到了这样的烦恼?手机里安卓系统的社交软件里,突然冒出了好多僵尸粉,这些家伙不仅不互动,还...
安卓平板电脑能刷系统,解锁无限... 你有没有想过,你的安卓平板电脑是不是也能来个“变身大法”,换换新系统呢?没错,今天就要来聊聊这个话题...
安卓双系统机器人,探索安卓双系... 你知道吗?在科技飞速发展的今天,手机已经不仅仅是一个通讯工具了,它还能变成一个多才多艺的小助手。最近...
ubuntu安装安卓系统下载,... 哇,你有没有想过,在你的Ubuntu电脑上安装一个安卓系统呢?想象你可以在同一个设备上无缝切换使用W...
安卓怎么进入闪氪系统,操作指南... 你有没有想过,你的安卓手机里隐藏着一个神秘的闪氪系统?别惊讶,这可不是科幻小说里的情节,而是真实存在...
如何关掉安卓系统通知,享受宁静 手机通知这东西,有时候真是让人又爱又恨。你有没有遇到过这种情况:手机里弹出来的通知一个接一个,让你分...
安卓系统为啥不更新了,安卓系统... 你有没有发现,最近你的安卓手机好像有点儿“懒”了,更新系统的时候总是慢吞吞的,甚至有时候直接告诉你:...
安卓系统手电在哪里打开,安卓手... 你有没有在安卓手机上找过手电筒功能,结果却像在茫茫大海中捞针一样?别急,今天就来给你详细揭秘,安卓系...
安卓系统个版本通用吗 你有没有想过,你的安卓手机上的系统版本是不是和别人的手机一样通用呢?这个问题听起来可能有点奇怪,但确...
安卓玩鸿蒙系统好吗,安卓用户玩... 你有没有想过,把安卓手机换成鸿蒙系统,会是怎样的体验呢?最近,这个话题在互联网上可是掀起了一阵热潮。...
安卓开源报名系统源码,架构设计... 你有没有想过,那些在手机上报名参加各种活动、课程或者考试的安卓应用,其实背后有一个神秘的“大脑”——...
峰米系统是安卓系统吗,揭秘安卓... 你有没有听说过峰米系统?最近这个话题在数码圈里可是挺火的。很多人都在问,峰米系统是安卓系统吗?今天,...
华为系统emui和安卓系统的区... 你有没有发现,手机的世界里,系统就像是它们的灵魂,决定了它们能跳得多高、飞得多远。今天,咱们就来聊聊...
朗逸安卓系统车载导航,安卓系统... 你有没有想过,开车的时候,导航系统就像是个贴心的导航小精灵,带你穿梭在城市的每一个角落?今天,就让我...
排球鹰眼系统辅助器安卓,安卓排... 你有没有想过,在观看排球比赛时,那些精准的鹰眼系统是如何让比赛画面更加清晰、精彩呢?现在,就让我带你...