---
license: apache-2.0
pipeline_tag: object-detection
---
# 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](https://arxiv.org/pdf/2406.03459v1).

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](https://huggingface.co./xbsu/LW-DETR/tree/main/pretrain_weights) to download.
- Release a series of pretrained models in LW-DETR. Please refer to [Hugging Face](https://huggingface.co./xbsu/LW-DETR/tree/main/pretrain_weights) to download.

## Catalogue