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

add hubconfig #436

Closed lyuwenyu closed 2 months ago

lyuwenyu commented 2 months ago 

import torch

models = torch.hub.list('lyuwenyu/RT-DETR')
print(models)

_ = torch.hub.load('lyuwenyu/RT-DETR', 'rtdetrv2_s', pretrained=False)
_ = torch.hub.load('lyuwenyu/RT-DETR', 'rtdetrv2_l')

_ = torch.hub.load('lyuwenyu/RT-DETR', 'rtdetrv2_r18vd', pretrained=False)
_ = torch.hub.load('lyuwenyu/RT-DETR', 'rtdetrv2_r34vd')
_ = torch.hub.load('lyuwenyu/RT-DETR', 'rtdetrv2_r50vd_m')
_ = torch.hub.load('lyuwenyu/RT-DETR', 'rtdetrv2_r50vd')

_ = torch.hub.load('lyuwenyu/RT-DETR', 'rtdetr_r18vd', pretrained=True)
_ = torch.hub.load('lyuwenyu/RT-DETR', 'rtdetr_r34vd', pretrained=False)
_ = torch.hub.load('lyuwenyu/RT-DETR', 'rtdetr_r50vd_m', pretrained=False)
_ = torch.hub.load('lyuwenyu/RT-DETR', 'rtdetr_r50vd', pretrained=False)