最近出现了更新的YOLO模型还有RT-DETR模型这是一个声称能击败YOLO模型的变换器模型我想将这些模型导出并进行比较并将它们添加到我的库中。

在这篇文章中我将带你了解如何将这些模型导出到ONNX并运行它们然后最终比较这些模型的速度。

将RT-DETR模型导出到ONNX

为了导出模型我们需要从模型的GitHub仓库克隆代码https://github.com/lyuwenyu/RT-DETR。

我将向你展示如何导出模型的第二个版本。

如果你想使用第一个版本导出步骤应该是类似的。

git

https://github.com/lyuwenyu/RT-DETR.git

然后我们需要创建一个Python虚拟环境来运行代码。

如果你已经有一个工作的环境那么你可以跳过这一步。

python

代码需要scipy库但它不在需求文件中你需要手动安装。

然后我们需要下载coco数据集的YAML配置文件和模型。

我们将在这里下载的模型是中等大小的模型。

你可以在这里找到其他模型的链接。

mkdir

https://github.com/lyuwenyu/storage/releases/download/v0.1/rtdetrv2_r34vd_120e_coco_ema.pth

wget

https://raw.githubusercontent.com/ultralytics/ultralytics/main/ultralytics/cfg/datasets/coco.yaml

现在我们可以导出模型了。

请注意如果你改变了模型大小那么你需要更改下面的配置文件路径以匹配模型。

python

configs/rtdetrv2/rtdetrv2_r34vd_120e_coco.yml

models/rtdetrv2_r34vd_120e_coco_ema.pth

--check

现在你有了你的模型在rtdetrv2_pytorch目录中的model.onnx文件。

你可以进入运行模型部分。

将YOLO-NAS导出到ONNX

为了导出YOLO-NAS模型你需要安装super_gradients库然后运行以下Python代码。

模型变体是YOLO_NAS_S、YOLO_NAS_M、YOLO_NAS_L。

from

super_gradients.common.object_names

import

pretrained_weightscoco)model.eval()

model.prep_model_for_conversion(input_size[1,

640,

现在你有了你的模型yolo_nas_s.onnx文件。

你可以进入运行模型部分。

将YOLOv10导出到ONNX

与模型的GitHub页面上的其他模型相比YOLOv10模型的延迟

我们需要克隆GitHub仓库https://github.com/THU-MIG/yolov10并创建一个Conda环境来运行导出代码。

git

https://github.com/THU-MIG/yolov10.git

yolov10conda

现在我们需要导出模型。

模型的前缀是jameslahm/支持的模型有yolov10n、yolov10s、yolov10m、yolov10b、yolov10l、yolov10x。

yolo

现在你有了你的模型yolov10目录中的yolov10m.onnx文件。

你可以进入运行模型部分。

将YOLOv9导出到ONNX

与模型的GitHub页面上的其他模型相比YOLOv9模型的参数数量

为了导出YOLOv9你需要有一个工作的正常安装的docker。

你可以启动一个容器并导出模型。

你可以在这里找到模型文件。

Create

https://github.com/WongKinYiu/yolov9.git

Now

nvcr.io/nvidia/pytorch:21.11-py3

Install

https://github.com/WongKinYiu/yolov9/releases/download/v0.1/yolov9-s.pt

Export

现在你有了你的模型yolov9目录中的yolov9-s.onnx文件。

你可以进入运行模型部分。

将YOLOv8导出到ONNX

将YOLOv8导出比其他模型更容易。

你只需要安装ultralytics并导出模型。

支持的模型有yolov8n、yolov8s、yolov8m、yolov8l和yolov8x。

如果你想的话你可以创建一个虚拟环境来隔离安装。

pip

https://github.com/ultralytics/assets/releases/download/v8.2.0/yolov8n.pt

yolo

现在你有了模型yolov8n.onnx文件可以进入运行模型部分。

运行模型

现在你有了你想要导出的ONNX模型你可以使用我编写的库来运行这些模型。

你可以在这里找到这个库。

库的README文件解释了如何链接到库。

以下是如何使用库中的RT-DETR模型的示例。

#include

回复“Python视觉实战项目”解锁31个超有趣的视觉项目大礼包本文仅供学习交流使用如有侵权请联系作者删除