The NumPy archive images are not RGB

google-deepmind / dsprites-dataset

Dataset to assess the disentanglement properties of unsupervised learning methods

Apache License 2.0

483 stars 69 forks source link

Closed psoulos closed 7 years ago

psoulos commented 7 years ago

The README states that the data is a NPZ NumPy archive with the following fields: imgs: (737280 x 3 x 64 x 64, uint8) Images in RGB.

However, the actual shape of the archive is (737280, 64, 64)

np.load('dsprites_ndarray_co1sh3sc6or40x32y32_64x64.npz')['imgs'].shape (737280, 64, 64)

Azhag commented 7 years ago

Thanks a lot for flagging, sorry about that!

I'll update the README now.