search

Megvii-BaseDetection / DenseTeacher

DenseTeacher: Dense Pseudo-Label for Semi-supervised Object Detection
Apache License 2.0
120 stars 12 forks source link

Using a different dataset, how to reset NUM_CLASSES? #29

Closed BroenLin closed 1 year ago

BroenLin commented 1 year ago

I am using a new dataset(not coco or voc) and the num_classes is 20 for my dataset instead of 80 for coco. I already changed num_classes in fcos.py. Yet, the config while training still shows that my model num_classes equals 80. config fcos plz help me.

ZRandomize commented 1 year ago

hi, please specify num_classes by insert "NUM_CLASSES=20" at here, i.e., MODEL.FCOS.NUM_CLASSES=20

BroenLin commented 1 year ago

hi, please specify num_classes by insert "NUM_CLASSES=20" at here, i.e., MODEL.FCOS.NUM_CLASSES=20

It works. Thanks a lot.