Hugging Face's logo

jozhang97
/
deta-swin-large

Object Detection
Transformers
PyTorch
Safetensors
deta
vision
Inference Endpoints
Model card Files Community
2
deta-swin-large / README.md
nielsr's picture
nielsr HF staff
Create README.md
001b61f
preview code
|
raw history blame contribute delete
No virus
642 Bytes
metadata
pipeline_tag: object-detection
tags:
  - vision

Detection Transformers with Assignment

By Jeffrey Ouyang-Zhang, Jang Hyun Cho, Xingyi Zhou, Philipp Krähenbühl

From the paper NMS Strikes Back.

TL; DR. Detection Transformers with Assignment (DETA) re-introduce IoU assignment and NMS for transformer-based detectors. DETA trains and tests comparibly as fast as Deformable-DETR and converges much faster (50.2 mAP in 12 epochs on COCO).