rtdetr_v2_r101vd / README.md
apolloparty's picture
Update README.md
a72a9e2 verified
---
license: apache-2.0
pipeline_tag: object-detection
library_name: transformers
---
Transformers usable lyuwenyu RT-DETRv2_101vd
Origin:
- https://github.com/lyuwenyu/RT-DETR/tree/main/rtdetrv2_pytorch
Converter:
- https://huggingface.co/jadechoghari/RT-DETRv2/tree/main
- Modification on config.json -> model_type to be usable with Transformers
Tested and working with:
- Transformers with RTDetrForObjectDetection, RTDetrImageProcessor, AutoModelForObjectDetection, AutoImageProcessor
- supervision with SAHI and annotators (except segmentation)
Code example (coming soon)
Source:
@misc{lv2023detrs,
title={DETRs Beat YOLOs on Real-time Object Detection},
author={Wenyu Lv and Shangliang Xu and Yian Zhao and Guanzhong Wang and Jinman Wei and Cheng Cui and Yuning Du and Qingqing Dang and Yi Liu},
year={2023},
eprint={2304.08069},
archivePrefix={arXiv},
primaryClass={cs.CV}
}
@misc{lv2024rtdetrv2improvedbaselinebagoffreebies,
title={RT-DETRv2: Improved Baseline with Bag-of-Freebies for Real-Time Detection Transformer},
author={Wenyu Lv and Yian Zhao and Qinyao Chang and Kui Huang and Guanzhong Wang and Yi Liu},
year={2024},
eprint={2407.17140},
archivePrefix={arXiv},
primaryClass={cs.CV},
url={https://arxiv.org/abs/2407.17140},
}