NUM_CLASSES on custom dataset ? about centermask2 HOT 4 CLOSED

I think you set them up properly.

Are there other problems?

When I use this project for face detection, it works well.

from centermask2.

It was actually working, it was a hyperparameter problem. Thank you for your answer!

from centermask2.

I'm trying to change the number of classes the network can detect. By using the exact same pipeline with another network I can get results but with CenterMask I get nothing at all. Looks like the network stays on the 80 COCO classes.

I've changed all the cfg keys containing NUM_CLASSES :

cfg.MODEL.ROI_HEADS.NUM_CLASSES = num_classes
cfg.MODEL.SEM_SEG_HEAD.NUM_CLASSES = num_classes
cfg.MODEL.RETINANET.NUM_CLASSES = num_classes
cfg.MODEL.FCOS.NUM_CLASSES = num_classes

I guess I'm missing something?

where is the cfg.MODEL.ROI_HEADS.NUM_CLASSES?? i can not find it can you help me ?

from centermask2.

It was actually working, it was a hyperparameter problem. Thank you for your answer!

hi ，how did you solve it, i met the similar problem

from centermask2.