Correlated Ellipses dataset for measuring disentanglement when the factors of variation are correlated
Switch branches/tags
Clone or download
Fetching latest commit…
Cannot retrieve the latest commit at this time.
Type Name Latest commit message Commit time
Failed to load latest commit information.

Correlated Ellipses Dataset

This repository contains the Correlated Ellipses dataset which can be used for testing disentanglement performance in the case when the factors of variation are not independent of each other.


Please check the Example.ipynb notebook which contains example code for sampling and conditional sampling from the dataset. Please note that the provided file is required for conditional sampling because it generates new samples on the fly. has the following requirements:

  • OpenCV
  • Numpy
  • Matplotlib

If you use this dataset, please consider citing.

  title={Hyperprior Induced Unsupervised Disentanglement of Latent Representations},
  author={Ansari, Abdul Fatir and Soh, Harold},
  booktitle={AAAI Conference on Artificial Intelligence},