search

facebookresearch / deepcluster

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

Training on Pascal VOC 2007 #20

Closed caojiawenn closed 5 years ago

caojiawenn commented 5 years ago

When training the model on Pascal VOC2007 for clustering, is the model randomly initialized ? or initialized with the parameters pre-trained on ImageNet?

mathildecaron31 commented 5 years ago

Hi,

We do not train the model on Pascal VOC2007 for clustering.

We pre-train a model with a joint clustering and classification objective on ImageNet without the labels. Then, in order to evaluate the quality of the features learned during this pre-training stage we use our weights as initialization to solve tasks (classification, object detection, segmentation) on Pascal VOC2007. In other words, the model is initialized with the weights that we leaned with deepcluster on ImageNet (without label).

Hope it helps