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

相关内容

热门资讯

安卓系统限制无法录音,探索无法... 你有没有遇到过这种情况?手机里明明装了录音软件,却突然发现,哎呀妈呀,竟然无法录音了!这可真是让人头...
怎么降级手机系统安卓,操作指南... 手机系统升级了,新功能层出不穷,但有时候,你可能会觉得,这系统太卡了,想回到那个流畅如丝的年代。别急...
米oa系统是安卓系统吗,深入解... 亲爱的读者,你是否曾好奇过,米OA系统是不是安卓系统的一员?这个问题,就像是一颗好奇的种子,悄悄地在...
手机刷安卓车载系统,手机刷机后... 你有没有发现,现在开车的时候,手机和车载系统之间的互动越来越紧密了呢?想象当你驾驶着爱车,一边享受着...
vivo安卓怎么降系统,viv... 手机用久了,是不是觉得系统越来越卡,运行速度大不如前?别急,今天就来教你怎么给vivo安卓手机降降级...
nova 4刷安卓系统,体验全... 最近手机界可是热闹非凡呢!听说华为nova 4要刷安卓系统了,这可真是让人兴奋不已。你有没有想过,你...
如果当初没有安卓系统,科技世界... 想象如果没有安卓系统,我们的生活会是怎样的呢?是不是觉得有点不可思议?别急,让我们一起穿越时空,探索...
安卓电视装win系统,系统转换... 亲爱的读者们,你是否曾想过,在你的安卓电视上装一个Windows系统,让它瞬间变身成为一台功能强大的...
安卓手机还原系统好处,重拾流畅... 你有没有遇到过安卓手机卡顿、运行缓慢的情况?别急,今天就来给你揭秘一下安卓手机还原系统的那些好处,让...
安卓系统能跑win吗,探索跨平... 你有没有想过,你的安卓手机里能不能装上Windows系统呢?这听起来是不是有点像科幻电影里的情节?别...
安卓车载系统蓝牙设置,畅享智能... 你有没有发现,现在开车的时候,手机和车载系统之间的互动越来越频繁了呢?这不,今天就来给你详细说说安卓...
奥利奥安卓系统,探索新一代智能... 你有没有想过,一块小小的奥利奥饼干竟然能和强大的安卓系统扯上关系?没错,今天就要来聊聊这个跨界组合,...
微信使用安卓系统,功能解析与操... 你有没有发现,现在用微信的人越来越多了呢?尤其是安卓系统的用户,简直就像潮水一样涌来。今天,就让我带...
体验最新原生安卓系统,极致体验... 你有没有想过,手机系统就像是我们生活的调味品,有时候换一种口味,生活都会变得有趣起来呢?最近,我体验...
安卓系统能玩原神,尽享奇幻冒险... 你有没有想过,在安卓系统上也能畅玩《原神》这样的热门游戏呢?没错,就是那个画面精美、角色丰富、玩法多...
安卓写手机银行系统,基于安卓平... 你有没有想过,手机银行系统在我们日常生活中扮演了多么重要的角色呢?每天刷刷手机,就能轻松管理账户,转...
僵尸之夜恐怖安卓系统,揭秘恐怖... 僵尸之夜,恐怖安卓系统来袭!想象一个寂静的夜晚,你正沉浸在美梦中,突然,一阵诡异的铃声打破了夜的宁静...
谷歌框架和安卓系统,构建智能移... 你有没有想过,为什么你的手机那么聪明,能帮你找到路线,还能帮你拍出美美的照片呢?这都要归功于一个超级...
安卓系统和oppo系统哪个流畅... 你有没有想过,手机系统哪个更流畅呢?安卓系统和OPPO系统,这两个名字听起来就让人心动。今天,咱们就...
安卓怎么用微软系统,利用微软系... 你是不是也和我一样,对安卓手机上的微软系统充满了好奇?想象那熟悉的Windows界面在你的安卓手机上...