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

相关内容

热门资讯

fedora9 安装-安装 F... 哎呀,说到安装Fedora9,这可真是个让人又爱又恨的小妖精!想象一下,你坐在电脑前,手里握着那张闪...
linux模拟器 window... 哎呀,说到在Windows上搞个Linux模拟器,我这心情真是激动得不行!你知道吗,有时候我就想在W...
switch支持破解序列号-S... 哎呀,今天咱们聊聊这个挺刺激的话题——Switch支持破解序列号。首先啊,我得说,这事儿可不是闹着玩...
电脑windows升级-电脑升... 哎,说到这个电脑升级,真的是让人又爱又恨呐!每次Windows一蹦出来那个升级提示,心里就嘀咕:“又...
电脑盘符不见了-电脑盘符消失怎... 天哪,今天打开电脑一看,怎么回事?我的D盘、E盘,还有那个存满美好回忆的F盘,都跑哪儿去了?这电脑是...
恢复丢失的分区-丢失的硬盘分区... 哎呀,真是吓死我了!那天,我一不小心,竟然把我的硬盘分区给弄丢了!你知道的,那些分区里可都是我的心血...
游戏中不起眼的后退键宏,为何能... 在游戏的世界里,每个按键都像是一个小小的咒语,而其中一个最让我心动的,就是那个不起眼的后退键宏。这不...
igfxtrayexe下载-电... 哎呀,说到这个Igfxtray.exe下载,我简直要抓狂了!你们知道吗,这个东西,简直就是电脑小白的...
passtemp是什么意思车上... 哎呀,说到这个Passtemp,真是让人一头雾水!我这开车的老司机,听说过各种车上的名词,什么ABS...
易分销下载:让小店生意飞起来的... 大家好,我是你们的小店主小明,今天我要跟大家聊聊那个让我的小店生意飞起来的神奇工具——易分销下载!说...
资源管理器 标签-资源管理器标... 哎呀,说到资源管理器里的标签功能,我这心里的五味瓶就打翻了。你知道吗,有时候它简直就是我的救星,有时...
android手机测试-And... 嘿,大家好!今天我要来聊聊我的Android手机测试经历,这可真是一场充满惊喜和惊吓的大冒险啊!首先...
冠心病及护理ppt-了解冠心病... 大家好,我是你们的心脏,今天我想和大家聊聊一个让我有点小紧张的话题——冠心病。这可不是什么小问题,它...
正版系统多次安装-正版系统频繁... 哎呀,每次说到这正版系统,我就一肚子火!你知道吗,我那心爱的电脑,被我装了又装,卸了又卸的正版系统,...
chrome os可以运行ap... 大家好,我是你们的小编,今天真是超级激动啊!你们知道吗?我们的ChromeOS现在居然可以运行APK...
网上竟有人公开贩卖身份证号大全... 哎呀我的天啊!今天我上网一看,发现竟然有人在网上公开贩卖什么“身份证号大全加手机号”,我真的是震惊了...
门诊医生工作站:充满挑战与戏剧... 哎呀,说到这门诊医生工作站,真是让人又爱又恨的地方!每天踏进这个小小的空间,就像是走进了一个充满挑战...
手动运行ghost-手动运行 ... 哎呀,说到手动运行Ghost,这可不是什么简单的事儿!我得告诉你,这就像是一场无人知晓的探险,每一步...
shopnc b2b2c 插件... 哎呀,说到这个ShopNCB2B2C插件啊,我这心里就激动得像小鹿乱撞!你知道吗,这个插件简直就是电...
联想u330p驱动-联想 U3... 大家好,我是一个普通的电脑用户,今天我要来聊聊那个让我又爱又恨的联想U330p驱动!说实话,每次想到...