NMS Strikes Back
Paper • 2212.06137 • Published • 1
How to use jozhang97/deta-resnet-50 with Transformers:
# Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("object-detection", model="jozhang97/deta-resnet-50") # Load model directly
from transformers import AutoModelForObjectDetection
model = AutoModelForObjectDetection.from_pretrained("jozhang97/deta-resnet-50", dtype="auto")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).