I know that I can simply switch num_queries to recognize 1 class, but in principle I don’t need classification at the moment, how can I turn it off? And does the absence of a classifier affect performance?

How to teach DETR to detect only BBox for one specific object, without classification

mahmutc August 22, 2024, 7:41am 2

I guess you can’t change the model but this one detects only one class:

Topic		Replies	Views
Fine-Tune DETR on custom dataset (less than 250 labels) Models	1	2108	February 6, 2024
Fine-tune DETR on custom dataset while keeping old class labels Models	0	127	July 24, 2024
Possible fix for trainer evaluation with object detection 🤗Transformers	0	318	February 7, 2024
Zero shot classification with manual pytorch Beginners	0	723	August 27, 2021
Example DeTr Object Detectors not predicting after fine tuning Beginners	7	1457	July 30, 2025