You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
I am also interested to see if there is an example on how to do this. I have a segmentation dataset with masks for a single class that I'd like to use as training data for this model. Please advise, thanks!
Hi @monocongo ,
I follow the COCO setting. If you can format your dataset in COCO format, you should be able to train it. But I haven't updated the code fo r a while. I am worried the new PyTorch code may break the code.
@chengyangfu How about the number of classes please? Inside the defaults.py folder, should '_C.MODEL.ROI_BOX_HEAD.NUM_CLASSES =81' be changed to reflect the exact number of classes for the custom dataset or should it be changed to 'Number of classes + 1'? where 1 = background class? In general, how do I handle number of classes when training on my own dataset?
❓ Questions and Help
The text was updated successfully, but these errors were encountered: