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稍微慢一点点。

相关内容

热门资讯

安卓系统对比骁龙,性能与生态的... 你有没有想过,为什么你的手机里装的是安卓系统,而不是苹果的iOS呢?又或者,为什么你的安卓手机里搭载...
qt程序安卓系统运行,基于Qt... 你有没有想过,为什么有些手机上的程序运行得那么顺畅,而有些却总是卡得让人抓狂?今天,就让我来给你揭秘...
安卓系统免费应用推荐,助你畅享... 手机里的应用是不是越来越多,有时候都挑花眼了呢?别急,今天我就来给你推荐一些安卓系统上的免费应用,让...
安卓系统视频通话app,打造无... 你有没有发现,现在手机上的视频通话功能越来越强大了?尤其是安卓系统上的那些视频通话app,简直让人爱...
安卓系统发现高危病毒,守护手机... 亲爱的手机用户们,最近可是有个大消息在安卓系统用户群里炸开了锅!没错,就是安卓系统发现了一款高危病毒...
安卓系统疯狂弹广告,揭秘广告软... 你有没有遇到过这种情况?手机里突然弹出一个广告,让你瞬间心情大崩溃?没错,说的就是安卓系统那让人头疼...
ebook 10进入安卓系统 你有没有发现,最近你的安卓手机里多了一个新伙伴——那就是电子书(ebook)10!没错,就是那个我们...
安卓系统如何调听筒,安卓系统调... 手机听筒声音突然变小了?别急,让我来教你如何轻松调教安卓系统的听筒,让它重新恢复活力!一、检查音量设...
安卓系统是怎么手机,解锁智能生... 你有没有想过,我们每天不离手的安卓手机,它背后的安卓系统究竟是怎么一回事呢?今天,就让我带你一探究竟...
安卓系统能代替windows系... 你有没有想过,我们日常使用的安卓系统和Windows系统,哪个才是真正的霸主呢?是不是有时候觉得安卓...
lp108安卓系统,功能特点与... 你有没有听说最近LP108安卓系统火得一塌糊涂?没错,就是那个让无数手机用户都为之疯狂的新系统!今天...
安卓系统挂载u盘,轻松实现数据... 你有没有想过,你的安卓手机或平板电脑突然变成了一个移动的U盘?没错,就是那种可以随意存取文件的神奇设...
i5 安卓系统,引领智能终端新... 你有没有想过,为什么你的手机总是卡得要命,而别人的手机却能流畅如丝?是不是因为你的手机搭载了那个传说...
安卓手机系统没有升级,揭秘潜在... 你有没有发现,你的安卓手机系统好像好久没升级了呢?是不是觉得有点out了?别急,今天就来给你详细聊聊...
安卓14系统定制v,创新功能与... 你知道吗?最近安卓系统又出新花样了!安卓14系统定制版V,这名字听起来就让人兴奋不已。今天,就让我带...
手机安卓系统越高越好,探索最新... 你有没有发现,每次手机更新系统,你的手机就像脱胎换骨了一样?没错,说的就是你,那个安卓手机!今天,咱...
鸿蒙系统怎么用回安卓,轻松实现... 你是不是也和我一样,对鸿蒙系统的新鲜感还没过,却又忍不住想回到熟悉的安卓世界?别急,今天就来手把手教...
苹果7跟安卓系统,性能对决与用... 你有没有想过,为什么苹果7那么受欢迎,而安卓系统却有着庞大的用户群体?今天,我们就来聊聊这个话题,看...
安卓手机刷简化系统,轻松实现流... 你有没有想过,你的安卓手机其实可以变得更加轻快、流畅呢?没错,就是通过刷简化系统!今天,就让我带你一...
社保掌上通安卓系统,轻松掌握在... 你有没有发现,现在的生活越来越离不开手机了?无论是购物、聊天还是办公,手机都能轻松搞定。这不,今天就...