v2.5.0
Datasets ⭐
EXAMPLE CODE
Wiki Art Dataset
Load Wiki Art Dataset with one line of code. Wiki Art is an art emotions dataset. Stream Wiki Art Dataset while training models in Pytorch and Tensorflow.
Visualization of wiki art dataset on the Activeloop Platform

Wiki Art dataset

What is Wiki Art Dataset?

The WikiArt dataset contains painting from 195 different artists. The dataset has 42129 images for training and 10628 images for testing. Because WikiArt is available to the public, it has a well-developed structure, WikiArt is often used in the field of machine learning. Namely, it is used to train AI on WikiArt data to discover its ability to recognize, classify, and generate art.

Download Wiki Art Dataset in Python

Instead of downloading the Wiki Art dataset in Python, you can effortlessly load it in Python via our open-source package Hub with just one line of code.

Load Wiki Art Dataset Training Subset in Python

1
import hub
2
ds = hub.load('hub://activeloop/wiki-art')
Copied!

Wiki Art Dataset Structure

Wiki Art Data Fields

  • images: tensor containing the image.
  • labels: tensor to represent various emotions.

Wiki Art Data Splits

How to use Wiki Art Dataset with PyTorch and TensorFlow in Python

Train a model on Wiki Art dataset with PyTorch in Python

Let's use Hub's built-in PyTorch one-line dataloader to connect the data to the compute:
1
dataloader = ds.pytorch(num_workers=0, batch_size=4, shuffle=False)
Copied!

Train a model on Wiki Art dataset with TensorFlow in Python

1
dataloader = ds.tensorflow()
Copied!

Additional Information about Wiki Art Dataset

Wiki Art Dataset Description

Wiki Art Dataset Curators

Art. Saif M. Mohammad and Svetlana Kiritchenko

Wiki Art Dataset Licensing Information

Hub users may have access to a variety of publicly available datasets. We do not host or distribute these datasets, vouch for their quality or fairness, or claim that you have a license to use the datasets. It is your responsibility to determine whether you have permission to use the datasets under their license.
If you're a dataset owner and do not want your dataset to be included in this library, please get in touch through a GitHub issue. Thank you for your contribution to the ML community!

Wiki Art Dataset Citation Information

1
@inproceedings{danielczuk2019segmenting,
2
title={Segmenting Unknown 3D Objects from Real Depth Images using Mask R-CNN Trained on Synthetic Data},
3
author={Danielczuk, Michael and Matl, Matthew and Gupta, Saurabh and Li, Andrew and Lee, Andrew and Mahler, Jeffrey and Goldberg, Ken},
4
booktitle={Proc. IEEE Int. Conf. Robotics and Automation (ICRA)},
5
year={2019}
6
}
Copied!

Wiki Art Dataset FAQs

What is the Wiki Art dataset for Python?

The WikiArt dataset is often used in the field of machine learning. Namely, it is used to train AI on WikiArt data to discover its ability to recognize, classify, and generate art. It contains painting from 195 different artists and the dataset has 42129 images for training and 10628 images for testing.

How can I use the Wiki Art dataset in PyTorch or TensorFlow?

You can stream the Wiki Art dataset while training a model in PyTorch or TensorFlow with one line of code using the open-source package Activeloop Hub in Python. See detailed instructions on how to train a model on Wiki Art dataset with PyTorch in Python or train a model on Wiki Art dataset with TensorFlow in Python.