search

Megvii-BaseDetection / DenseTeacher

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

在用一个新的数据集。如何改变num_classes? #30

Closed BroenLin closed 1 year ago

BroenLin commented 1 year ago

我在用一个遥感数据集，num_classes只有20，但是config设置里num_classes是80，对应coco数据集，请问如何更改。我已经对fcos.py进行了更改，但是不起作用。训练时的config仍然是80。虽然可以训练到model_iter_0200000.pth，但是仍旧因为category_id=56而报错。 fcos

config

ZRandomize commented 1 year ago

请更改config.py，而不是直接更改code