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.61k stars 303 forks source link

Loading model from a .pth file #445

Open isuchy opened 2 months ago

isuchy commented 2 months ago

Hi there,

sorry for a newbie question, but I can't find the answer anywhere. Could anyone advise me how to load a RT-DETR model from .pth checkpoint file for use in object detection? In the README md this information is missing, there is only a brief description of how to train, eval and export the model, see screenshot.

image

Thanks in advance.

lyuwenyu commented 2 months ago

You can see these files

https://github.com/lyuwenyu/RT-DETR/blob/main/hubconf.py https://github.com/lyuwenyu/RT-DETR/blob/main/rtdetrv2_pytorch/references/deploy/rtdetrv2_torch.py