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

相关内容

热门资讯

安卓高系统怎么用美化,打造专属... 亲爱的安卓用户们,你是不是也和我一样,对手机系统美化情有独钟呢?想要让你的安卓手机焕然一新,变得个性...
安卓系统怎么开夜间模式,安卓系... 亲爱的手机控们,你是不是在夜晚使用安卓手机时,眼睛感到有些不适?别担心,今天我要给你揭秘一个超级实用...
王者安卓系统用苹果人脸,一场视... 你知道吗?最近在手机圈里可是掀起了一股不小的波澜呢!那就是王者安卓系统竟然用上了苹果人脸识别技术!是...
安卓444怎么升级系统,轻松迈... 你那安卓444的小家伙是不是已经有点儿落伍了?别急,今天就来给你详细说说怎么给它来个系统升级,让它焕...
安卓系统raw修图软件,探索安... 你有没有发现,手机拍照越来越方便了,但有时候拍出来的照片还是不够完美呢?别急,今天就来给你安利几款安...
安卓系统的王者切换苹果,从安卓... 你知道吗?最近身边的朋友圈里掀起了一股热潮,那就是安卓系统的王者们纷纷切换到苹果阵营。这可真是让人大...
安卓用按钮退出系统,Andro... 你有没有发现,现在用安卓手机的时候,退出系统的方式好像有点变化呢?是不是有时候按了按钮,系统还是纹丝...
安卓系统怎么删除相机,安卓系统... 手机里的照片越来越多,是不是感觉相机里都快装不下你的美好回忆了?别急,今天就来教你怎么轻松地在安卓系...
安卓双系统点歌机系统包,打造智... 你有没有想过,在安卓双系统点歌机里,那些美妙的音乐是如何被唤醒的呢?今天,就让我带你一探究竟,揭开安...
安卓不卡顿的系统,安卓系统不卡... 你有没有想过,为什么有些安卓手机用起来就是那么流畅,而有些却总是卡卡卡个不停?今天,就让我带你一探究...
鸿蒙系统切换安卓死机,系统兼容... 最近手机界可是热闹非凡呢!听说华为的新操作系统鸿蒙系统上线了,不少小伙伴都跃跃欲试,想要体验一下这个...
系统能被卸载吗安卓,卸载与优化... 你有没有想过,你的安卓手机里的系统,是不是也能像卸载一个APP那样简单呢?今天,就让我带你一探究竟,...
安卓永远超不过苹果系统,谁将永... 你有没有想过,为什么安卓手机再怎么努力,好像永远都追不上苹果系统的光芒呢?这背后,可是有着不少门道呢...
安卓系统手机位置共享,探索安卓... 你有没有想过,当你的朋友突然告诉你:“我快到你了,你定位一下我哦!”这时,你的安卓手机位置共享功能就...
安卓系统怎么添加表情包,表情包... 你是不是也和我一样,手机里收藏了超多搞笑的表情包,想要在聊天的时候分享给朋友,但又不知道怎么在安卓系...
安卓变黑鲨系统下载,深度解析系... 你有没有发现,你的安卓手机最近有点儿不一样了?是不是觉得屏幕上的图标变得酷炫了,操作起来也更加流畅了...
windows平板如何装安卓系... 亲爱的电脑迷们,你是否曾梦想过在你的Windows平板上体验安卓系统的流畅与丰富应用?别再羡慕那些安...
鸿蒙系统如何恢复到安卓系统中,... 你是不是也和我一样,对鸿蒙系统到安卓系统的转换感到好奇呢?听说最近很多人都在尝试将手机从鸿蒙系统切换...
苹果手机升级安卓系统,探索安卓... 你有没有想过,如果有一天苹果手机突然宣布要升级成安卓系统,那会是怎样的景象呢?想象那些习惯了iOS操...
iphone能不能安卓系统,探... 最近手机圈可是热闹非凡呢!不少朋友都在问我:“iPhone能不能用安卓系统?”这个问题,真是让人头都...