Add Frey-face dataset, used e.g. by VAE paper.

Author: lucasb-eyer

DeepFried2/datasets/__init__.py

@@ -1,6 +1,7 @@
 from . import mnist
 from . import cifar10
 from . import cifar100
+from . import freyface
 from . import sequence_sum
 from . import stanfordbg
 from . import svhn2

DeepFried2/datasets/freyface.py

@@ -0,0 +1,13 @@
+from DeepFried2 import floatX as _fX
+from DeepFried2.zoo.download import download as _download
+
+import scipy.io as _sio
+
+
+def data():
+    fname = _download('http://www.cs.nyu.edu/~roweis/data/frey_rawface.mat')
+    mat = _sio.loadmat(fname)
+
+    X = mat['ff'].T.reshape(-1, 28, 20).astype(_fX)
+    X /= 255
+    return X