Hugging Face's logo

jozhang97
/
deta-resnet-50-24-epochs

Object Detection
Transformers
PyTorch
Safetensors
deta
vision
Inference Endpoints
Model card Files Community
1
deta-resnet-50-24-epochs / README.md
nielsr's picture
nielsr HF staff
Create README.md
fb0fbdd
preview code
|
raw
history blame contribute delete
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).