search

vturrisi / solo-learn

solo-learn: a library of self-supervised methods for visual representation learning powered by Pytorch Lightning
MIT License
1.42k stars 185 forks source link

Custom Dataset object detection #355

Closed kalikhademi closed 1 year ago

kalikhademi commented 1 year ago

Hi,

I have pretrained a SSL method using main_pretrain.py script. I obtained the checkpoint and I would like to use it to detect objects on my custom dataset. Could you please let me know how to register my custom dataset to work with train_object_detection.py. My dataset has coco format json files for both train and valid.

vturrisi commented 1 year ago

Hey @kalikhademi. The object detection code is taken from the original MoCo GitHub repo and slightly adapted. It relies heavily on detectron as well, so I suggest you to check detectron directly.