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

关于trt模型推理 #279

Closed NeKoooo233 closed 4 months ago

NeKoooo233 commented 5 months ago

那个,我现在导出了trt模型,按照readme里面的教程,可以使用trtinfer.py进行推理,尽管这非常简单,我尝试了将文件中的engine_path改为模型路径,并运行python trtinfer.py(大概不是这么用的),但是并没有起作用,我想知道怎么使用trtinfer.py =.=

lyuwenyu commented 5 months ago
blob = {
    'images': im
    'orig_target_sizes': orig_size,
}

m = TRTInference('path/to/engine')
r = m(blob)