pytorch 41 yolov8的无nsm后处理的onnxruntime部署方案_yolov8 onnx 添加后处理和nms

作者：Li_阴宅 | 2024-08-14 15:05:06

踩

yolov8 onnx 添加后处理和nms

自onnx11版本及以后，onnx支持将nms操作嵌入到模型中。本博文将yolov8模型的nms操作嵌入到模型中可以大幅度的简化模型部署代码，提升模型推理速度（在以往大家都是使用opencv的库进行nms操作，CPU效率必然是不如GPU的）。同时修改了yolov8输出的格式，移除了每一个类别的conf信息，只保留了maxClassConf和maxClassId。
后续会补充C++、C#部署案例，目前仅有python onnxruntime部署案例。

1、模型结构分析

1.1 onnx模型导出

使用以下代码即可将yolov8模型导出为onnx文件

from ultralytics import YOLO

model = YOLO("yolov8s.pt")  # load a pretrained model (recommended for training)
path = model.export(1
2
3

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/Li_阴宅/article/detail/979861