search

lightly-ai / lightly

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

How to use lightly models on custom datasets #1516

Closed fereshtehforghani closed 3 months ago

fereshtehforghani commented 3 months ago

I have a custom dataset that is inherited from datasets.ImageNet. Should I make this a LightlyDataset for my models to work? if yes how?

My dataset looks something like this: class MyDataset(datasets.ImageNet):

guarin commented 3 months ago

Hi, you don't need a LightlyDataset, any torch dataset will work :)

guarin commented 3 months ago

I'll close this for now, please reopen or make a new issue if you have more questions.