discobox for custom class of 1 class. in the mask_feat_head, the item of num_classes should be 1 or 2 ? Does this value need to be changed?

LiWentomng / BoxInstSeg

A toolbox for box-supervised instance segmentation.

Apache License 2.0

405 stars 38 forks source link

discobox for custom class of 1 class. in the mask_feat_head, the item of num_classes should be 1 or 2 ? Does this value need to be changed? #15

Open ssli23 opened 1 year ago

LiWentomng commented 1 year ago

@ssli23 Sorry to reply later. The num_classes in the config is the actual category number, which need not add 1. So, num_classesshould be 1 for custom class of 1 class.