facebookresearch / deepcluster

Deep Clustering for Unsupervised Learning of Visual Features
Other
1.69k stars 325 forks source link

Training dataset contains both train and test images #60

Closed amajee11us closed 4 years ago

amajee11us commented 4 years ago

Hi.. Great Work !! I was wondering if the data used for training the deepcluster pseudo-labels use both the train and test images? If so how would be the folder structure for the dataset?

mathildecaron31 commented 4 years ago

Hi, thank you for your interest in our work ! We use only images from the train split of ImageNet in our training. But you can choose to train the model on any data you want. Hope it helps.