search

hmmlillian / Gray-Image-Retrieval

The source code of "Color Reference Recommendation" in the paper "Deep Exemplar-based Colorization".
https://arxiv.org/pdf/1807.06587.pdf
46 stars 8 forks source link

how to create the dataset #3

Open cccusername opened 4 years ago

cccusername commented 4 years ago

hello, @hmmlillian . If possible, could your share the code to creaate the dataset? Thank you!

hmmlillian commented 4 years ago

@cccusername We randomly sampled 700 classes from ImageNet as described in paper (animals (15%), plants (15%), people (20%), scenery (25%), food (5%), transportation (15%) and artifacts (5%)). For each class, we randomly 1,000 sample image pairs with different extents of similarity (45% of Top-5 similarity searched used the code of Gray-Image-Retrieval, 45% of randomly sampled within the same class, and 10% of randomly sampled within the same category). The major part is the image retrieval as we shared in this repo.