文章预览
↑ 点击 蓝字 关注极市平台 作者丨陈er 来源丨GiantPandaCV 编辑丨极市平台 极市导读 本文采用RT-DETR两种不同风格的onnx格式，使用onnxruntime20行代码，无需nms操作即可实现简易部署推理。 >> 加入极市CV技术交流群，走在计算机视觉的最前沿 【前言】 RT-DETR是由百度近期推出的DETR-liked目标检测器，该检测器由HGNetv2、混合编码器和带有辅助预测头的Transformer编码器组成，整体结构如下所示。 本文将采用RT-DETR两种不同风格的onnx格式，使用onnxruntime20行代码，无需nms操作即可实现简易部署推理. 一、原生onnx+ort推理方式 使用以下命令抽取出模型配置文件和模型参数文件： python tools/export_model.py -c configs/rtdetr/rtdetr_hgnetv2_l_6x_coco.yml -o weights=https://bj.bcebos.com/v1/paddledet/models/rtdetr_hgnetv2_l_6x_coco.pdparams trt=True --output_dir=output_inference 转化模型为onnx形式： paddle
………………………………
