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

backbone问题 #263

Open zhangsgi opened 7 months ago

zhangsgi commented 7 months ago

你好，我看paddle里面有HGNetv2的backbone，rtdetr_paddle/ppdet/modeling/backbones/hgnet_v2.py。这个文件能直接应用于pytorch里面吗？如果不可以，后期会考虑出pytorch版本的HGNetv2吗？

lyuwenyu commented 7 months ago

不能直接在torch里用， 如果需要的话 后期可以加一下

zhangsgi commented 7 months ago

有需要，希望作者能把pytorch版本的HGNetv2安排上日程，十分感谢！

Peterande commented 6 months ago

有需要，希望作者能把pytorch版本的HGNetv2安排上日程，十分感谢！

Pytorch可以用这个https://huggingface.co/timm/hgnetv2_b4.ssld_stage2_ft_in1k 对应文章中的L