Home

Awesome

ImageNet-Sketch

ImageNet-Sketch

Description

ImageNet-Sketch data set consists of 50000 images, 50 images for each of the 1000 ImageNet classes. We construct the data set with Google Image queries "sketch of __", where __ is the standard class name. We only search within the "black and white" color scheme. We initially query 100 images for every class, and then manually clean the pulled images by deleting the irrelevant images and images that are for similar but different classes. For some classes, there are less than 50 images after manually cleaning, and then we augment the data set by flipping and rotating the images.

This github repository consists of the scripts we used to conduct query and clean images.

Download the Data

ImageNet-Sketch Leaderboard

MethodReferenceBackboneFrom Scratch*Top1-Acc.Top5-Acc.
Texture Debiased AugmentationHermann et al. (NeurIPS 2020)ResNet50Y30.9%51.4%
Anisotropic diffusionMishra et al.ResNet50Y24.49%41.81%
Random ConvolutionsXu et al.AlexNetY18.09%35.40%
RSCHuang et al. (ECCV 2020)AlexNetY16.12%30.78%
PARWang et al. (NeurIPS 2019)AlexNetN13.06%26.27%
AlexNet BaselineAlexNetN/A12.04%24.80%

*This column indicates whether the model is trained from the scratch or built and fine-tuned based a pretrained backbone model.

Analysis

Reference

The data set is introduced together with the following paper in NeurIPS 2019, so if you find this data set helpful, please consider citing it:

Learning Robust Global Representations by Penalizing Local Predictive Power

@inproceedings{wang2019learning,
        title={Learning Robust Global Representations by Penalizing Local Predictive Power},
        author={Wang, Haohan and Ge, Songwei and Lipton, Zachary and Xing, Eric P},
        booktitle={Advances in Neural Information Processing Systems},
        pages={10506--10518},
        year={2019}
}