lightly-ai / lightly

A python library for self-supervised learning on images.
https://docs.lightly.ai/self-supervised-learning/
MIT License
3.13k stars 276 forks source link

Any plans for self-supervised object detection? #936

Closed faris-k closed 1 year ago

faris-k commented 2 years ago

Seeing as lightly has a ton of great models for self-supervised representation learning for "classification tasks," are there any plans in the future to implement methods for self-supervised object detection? Here's a link to a survey article and a few noteworthy papers in the area.

I understand if end-to-end object detection methods are beyond the scope of lightly right now, but I'd figure I'd ask since it might be cool to see more of these types of models on lightly 😃

guarin commented 2 years ago

Hi @faris-k,

We are very much interested in self-supervised object detection and pretraining for object detection. We did not yet implement any methods because there does not seem to be a well defined standard on how to train such networks and how to do the data loading. This is in contrast with image classification where most models follow more or less the same training setup with only minor differences in the model architecture. We would love to explore self-supervised learning for object detection more in detail but it is currently not a priority.

Note that better image classification pretraining often also increases performance on object detection tasks. If you are interested in using a model available in lightly for training an object detector then please have a look at our Detectron2 Tutorial