How to retrieve similar reference image for other datasets?

[htzheng]

I want to train CoCosNet for other datasets, however, I'm not sure how the similar reference image for ADE20K dataset (ade20k_ref.txt) is generated. Could you elaborate on it more? Thanks!

[panzhang0212]

If your dataset is face dataset, you could random choice reference. If your dataset is complex(e.g ADE20k), you maybe need to retrieve reference.
How to retrieve:
1.pretrained vgg model to extract feature(I use the feature after max pool before FC layer). Then you can get feature dictionary
2.for every image, you need to calculate feature distance(I use cos distance) between this feature and other features, then choose the top K ref image.