| 9 | from lib.config import data_dir |
| 10 | |
| 11 | def faces(ntrain=None, nval=None, ntest=None, batch_size=128): |
| 12 | path = os.path.join(data_dir, 'faces_364293_128px.hdf5') |
| 13 | tr_data = H5PYDataset(path, which_sets=('train',)) |
| 14 | te_data = H5PYDataset(path, which_sets=('test',)) |
| 15 | |
| 16 | if ntrain is None: |
| 17 | ntrain = tr_data.num_examples |
| 18 | if ntest is None: |
| 19 | ntest = te_data.num_examples |
| 20 | if nval is None: |
| 21 | nval = te_data.num_examples |
| 22 | |
| 23 | tr_scheme = ShuffledScheme(examples=ntrain, batch_size=batch_size) |
| 24 | tr_stream = DataStream(tr_data, iteration_scheme=tr_scheme) |
| 25 | |
| 26 | te_scheme = SequentialScheme(examples=ntest, batch_size=batch_size) |
| 27 | te_stream = DataStream(te_data, iteration_scheme=te_scheme) |
| 28 | |
| 29 | val_scheme = SequentialScheme(examples=nval, batch_size=batch_size) |
| 30 | val_stream = DataStream(tr_data, iteration_scheme=val_scheme) |
| 31 | return tr_data, te_data, tr_stream, val_stream, te_stream |