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和安卓的操作系统,还有个神秘的鸿蒙系统?今天,咱们就来聊...
哪个安卓机系统好用,探索安卓系... 你有没有想过,手机里的安卓系统就像是个大厨,不同的系统就像不同的烹饪手法,有的让你吃得津津有味,有的...
安卓如何控制苹果系统,从安卓到... 你知道吗?在这个科技飞速发展的时代,安卓和苹果两大操作系统之间的较量从未停歇。虽然它们各自有着忠实的...
安卓原生系统文件夹,安卓原生系... 你有没有发现,每次打开安卓手机,里面那些文件夹就像是一个个神秘的宝箱,里面藏着各种各样的宝贝?今天,...
基于安卓系统的游戏开发,从入门... 你有没有想过,为什么安卓手机上的游戏总是那么吸引人?是不是因为它们就像是你身边的好朋友,随时随地都能...
安卓系统怎样装驱动精灵,安卓系... 你那安卓设备是不是突然间有点儿不给力了?别急,今天就来手把手教你如何给安卓系统装上驱动精灵,让你的设...
如何本地安装安卓系统包,详细步... 你有没有想过,把安卓系统装在你的电脑上,是不是就像给电脑穿上了时尚的新衣?想象你可以在电脑上直接玩手...
安卓12卡刷系统教程,体验全新... 你有没有发现,你的安卓手机最近有点儿不给力了?运行速度慢得像蜗牛,是不是也想给它来个“换血大法”,让...
安卓系统无法打开swf文件,安... 最近是不是发现你的安卓手机有点儿不给力?打开SWF文件时,是不是总是出现“无法打开”的尴尬局面?别急...
鸿蒙系统依赖于安卓系统吗,独立... 你有没有想过,我们手机里的那个鸿蒙系统,它是不是真的完全独立于安卓系统呢?这个问题,估计不少手机控都...
适合安卓系统的图片软件,精选图... 手机里堆满了各种美美的照片,是不是觉得找起来有点头疼呢?别急,今天就来给你安利几款超级适合安卓系统的...
阴阳师安卓系统典藏,探寻阴阳师... 亲爱的阴阳师们,你是否在安卓系统上玩得如痴如醉,对那些精美的典藏式神们垂涎欲滴?今天,就让我带你深入...
安卓系统有碎片化缺点,系统优化... 你知道吗?在手机江湖里,安卓系统可是个响当当的大侠。它那开放、自由的个性,让无数手机厂商和开发者都为...
安卓4系统手机微信,功能解析与... 你有没有发现,现在市面上还有很多安卓4系统的手机在使用呢?尤其是那些喜欢微信的朋友们,这款手机简直就...
鸿蒙系统是安卓的盗版,从安卓“... 你知道吗?最近在科技圈里,关于鸿蒙系统的讨论可是热闹非凡呢!有人说是安卓的盗版,有人则认为这是华为的...
安卓系统怎么剪辑音乐,轻松打造... 你是不是也和我一样,手机里存了超多好听的歌,但是有时候想给它们来个变身,变成一段专属的旋律呢?别急,...
怎么把安卓手机系统变为pc系统... 你有没有想过,把你的安卓手机变成一台PC呢?听起来是不是有点酷炫?想象你可以在手机上玩电脑游戏,或者...
手机怎么装安卓11系统,手机安... 你有没有想过,让你的手机也来个“青春焕发”,升级一下系统呢?没错,就是安卓11系统!这个新系统不仅带...
安卓系统如何拼网络,构建高效连... 你有没有想过,你的安卓手机是怎么和网络“谈恋爱”的呢?没错,就是拼网络!今天,就让我带你一探究竟,看...
安卓系统怎么看小说,轻松畅享电... 你有没有发现,手机里装了那么多应用,最离不开的竟然是那个小小的小说阅读器?没错,就是安卓系统上的小说...