search

lyuwenyu / RT-DETR

[CVPR 2024] Official RT-DETR (RTDETR paddle pytorch), Real-Time DEtection TRansformer, DETRs Beat YOLOs on Real-time Object Detection. 🔥 🔥 🔥
Apache License 2.0
2.31k stars 258 forks source link

onnx COCO数据集 Map情况 #427

Open guwenjie0730 opened 1 month ago

guwenjie0730 commented 1 month ago

嗨，文玉姐姐， 可以麻烦出一个coco val2017数据集上 测试onnx model 的map代码吗

lyuwenyu commented 1 month ago

改造一下这个函数，把里面的torch-model推理的逻辑 改成onnx-model就行

https://github.com/lyuwenyu/RT-DETR/blob/55ca98fee1703924a846adcb7808440312c34229/rtdetrv2_pytorch/src/solver/det_engine.py#L107

( onnx推理参考 https://github.com/lyuwenyu/RT-DETR/blob/main/rtdetrv2_pytorch/references/deploy/rtdetrv2_onnxruntime.py#L30