RF-DETR Num_classes mismatch warning

🤝 Community Help
1

Hi! I’m getting this warning when evaluating RF-DETR:

num_classes mismatch: pretrain weights has 0 classes, but your model has 1 classes reinitializing detection head with 0 classes

when I pass in the checkpoint_best_ema.pth from my training as the pretrained weights. But my model seems to be performing quite well. What might be the issue, “pretrain weights has 0 classes” is very concerning?

2

Hi @mila88!
Unfortunately I am unable to support local evaluation of RF-DETR. Apologies for the inconvenience!!

3

This topic was automatically closed 21 days after the last reply. New replies are no longer allowed.