Was anyone able to train on their own dataset without labels?

wvangansbeke / Unsupervised-Classification

SCAN: Learning to Classify Images without Labels, incl. SimCLR. [ECCV 2020]

https://arxiv.org/abs/2005.12320

Other

1.37k stars 268 forks source link

Was anyone able to train on their own dataset without labels? #111

Open Harsh9524 opened 2 years ago

Harsh9524 commented 2 years ago

I went through the #8 and #19 and read the discussion about training the model on custom dataset. If anyone has successfully trained on custom dataset can they share the repository?

edwardcho commented 2 years ago

Hello Sir,

I am looking forward answers for this issue.

Thanks, Edward Cho.

Mohit-robo commented 1 year ago

Yes, you would need to create a Dataloader for your dataset, In my case, I had 2 classes, and create a Dataloader, that's it.

itsMohitShah commented 1 year ago

Hi Mohit, Can you share the code for that?

I believe you are talking about making your own .py file in data folder (data/custom_data.py) Along with that, could you also please share your version of common_confi? Will be a massive help.