facebookresearch/dino

Number of classes

VGrondin opened this issue · 1 comments

Hello, thank you for your work!
I have a small datsaset containing only one class of instances (trees). I looked in the code and it seems like the number of classes in the VIsionTransformer is always zero (num_classes=0). Is this normal? I am not sure I understand the difference between the num_classes and num_labels used in the eval script.

Thank you

I think the number of classes you have to try is 1 greater then that of orignal number of label like for you it will be 2 (number of classes ) dn_labblebook or so parameter must be = or > number of classes