dSprites Dataset

Estimated reading: 2 minutes 1545 views

Visualization of the dSprites dataset in the Deep Lake UI

dSprites dataset

What is dSprites Dataset?

The dSprites (Disentanglement testing Sprites) dataset is created to assess the disentanglement properties of unsupervised learning methods. It can be used to evaluate how well models recover the ground truth latents. The dataset contains 737280 images of 64×64 resolution. The dataset comes along with latents values, and classes.

Download dSprites Dataset in Python

Instead of downloading the dSprites dataset in Python, you can effortlessly load it in Python via our Deep Lake open source with just one line of code.

Load dSprites Dataset in Python

				
					import deeplake
ds = deeplake.load('hub://activeloop/dsprites')

dSprites Dataset Structure

Data Fields

images: tensor containing black and white images
latents_classes: tensor containing the index of the latents factor values
latents_values: tensor containing values of the latents factors

How to use dSprites Dataset with PyTorch and TensorFlow in Python

Train a model on dSprites dataset with PyTorch in Python

Let’s use Deep Lake built-in PyTorch one-line dataloader to connect the data to the compute:

				
					dataloader = ds.pytorch(num_workers=0, batch_size=4, shuffle=False)

Train a model on dSprites dataset with TensorFlow in Python

				
					dataloader = ds.tensorflow()

Additional Information about dSprites Dataset

dSprites Dataset Description

Repository: https://github.com/deepmind/dsprites-dataset
Paper: Matthey, L., Higgins, I., Hassabis, D., & Lerchner, A. (2017). dsprites: Disentanglement testing sprites dataset.
Activeloop Deep Lake: https://app.activeloop.ai/activeloop/dsprites

dSprites Dataset Contributors

Matthey, L., Higgins, I., Hassabis, D., & Lerchner, A.

dSprites Dataset Licensing Information

Deep Lake 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!

dSprites Dataset Citation Information

				
					@misc{dsprites17,
author = {Loic Matthey and Irina Higgins and Demis Hassabis and Alexander Lerchner},
title = {dSprites: Disentanglement testing Sprites dataset},
howpublished= {https://github.com/deepmind/dsprites-dataset/},
year = "2017",
}