NVlabs / FreeSOLO

FreeSOLO for unsupervised instance segmentation, CVPR 2022
Other
312 stars 32 forks source link

How to train own custom dataset? #12

Open xiaonanQua opened 1 year ago

xiaonanQua commented 1 year ago

Hello, I have read your paper and code, and I would like to ask how to train my own dataset. My dataset is in the category of defect detection, and I can hardly find similar images in the COCO dataset. Following your instructions, I downloaded the pre-trained model DenseCL and used inference_freemask.sh to generate coarse masks, but the generated masks hardly mark the defective areas. Therefore, I would like to ask if I need to train my own DenseCL model to generate coarse masks?