what does the memory data loader do in the basic_template.py?

Hzzone / ProPos

Self-Supervised Learning for Deep Clustering (TPAMI 2022)

104 stars 15 forks source link

what does the memory data loader do in the basic_template.py? #16

Closed ydzhang12345 closed 1 year ago

ydzhang12345 commented 1 year ago

I am trying to use ProPos on my custom unlabeled data. I believe I will need to rewrite basic_template.py. Can you explain a bit what deos the memory loader do?

Hzzone commented 1 year ago

The memory data loader should be used to extract the features for pseudo-labeling without any strong data augmentation. If you need to train on your own unlabeled data, you can refer to https://github.com/Hzzone/ProPos/issues/7. You can also organize the dataset like ImageNet by specifying the data_root.