LW-DETR: A Transformer Replacement to YOLO for Real-Time Detection

This is the official repository with PyTorch implementation of LW-DETR: A Transformer Replacement to YOLO for Real-Time Detection.

The code is at https://github.com/Atten4Vis/LW-DETR

☀️ If you find this work useful for your research, please kindly star our repo and cite our paper! ☀️

Highlights

• Release a series of real-time detection models in LW-DETR, including LW-DETR-tiny, LW-DETR-small, LW-DETR-medium, LW-DETR-large and LW-DETR-xlarge, named <LWDETR_*size_60e_coco.pth>. Please refer to Hugging Face to download.
• Release a series of pretrained models in LW-DETR. Please refer to Hugging Face to download.

Catalogue