Skip to content

liaopeiyuan/artbench

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

19 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

ArtBench

PyTorch Usage

The ArtBench Dataset: Benchmarking Generative Models with Artworks
Peiyuan Liao*, Xiuyu Li*, Xihui Liu, Kurt Keutzer
* 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:

  • it is class-balanced while most previous artwork datasets suffer from the long tail class distributions
  • the images are of high quality with clean annotations
  • it is created with standardized data collection, annotation, filtering, and preprocessing procedures.

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}
}

Releases

No releases published

Languages