Home

Awesome

ArtBench

PyTorch Usage

Preprint | Kaggle | Papers With Code

The ArtBench Dataset: Benchmarking Generative Models with Artworks<br/> Peiyuan Liao*, Xiuyu Li*, Xihui Liu, Kurt Keutzer<br/> * equal contribution

ArtBench-10 is the first class-balanced, high-quality, cleanly annotated, and standardized dataset for benchmarking artwork generation. It comprises 60,000 images of artwork from 10 distinctive artistic styles, with 5,000 training images and 1,000 testing images per style.

ArtBench-10 has several advantages over previous artwork datasets:

We provide three versions of the dataset with different resolutions (32 x 32, 256 x 256, and original image size), formatted in a way that is easy to be incorporated by popular machine learning frameworks.

Accessing Dataset

Using

See artbench.py for PyTorch usage. You only need ~20 lines of code to start using ArtBench-10 in your PyTorch workloads!

PyTorch Usage

Citation

If you find the work useful in your research, please consider citing:

@article{liao2022artbench,
  title={The ArtBench Dataset: Benchmarking Generative Models with Artworks},
  author={Liao, Peiyuan and Li, Xiuyu and Liu, Xihui and Keutzer, Kurt},
  journal={arXiv preprint arXiv:2206.11404},
  year={2022}
}