diff options
| author | Jules Laplace <julescarbon@gmail.com> | 2019-12-10 22:24:21 +0100 |
|---|---|---|
| committer | Jules Laplace <julescarbon@gmail.com> | 2019-12-10 22:24:21 +0100 |
| commit | 3852f8b30183f87633151aab017d7fbe4891bdd6 (patch) | |
| tree | 2759c88fa7d540bf36d7d4b542ca17d2fcd1ef72 /cli/app/commands | |
| parent | 0458542e4d06ae7aaae23c15e04ef43f54ad4f8d (diff) | |
refactor and add hdf5 support
Diffstat (limited to 'cli/app/commands')
| -rw-r--r-- | cli/app/commands/biggan/search.py | 7 |
1 files changed, 3 insertions, 4 deletions
diff --git a/cli/app/commands/biggan/search.py b/cli/app/commands/biggan/search.py index f1cf385..0b31a0d 100644 --- a/cli/app/commands/biggan/search.py +++ b/cli/app/commands/biggan/search.py @@ -43,7 +43,6 @@ def cli(ctx, opt_fp_in, opt_dims, opt_video): input_trunc = inputs['truncation'] output = generator(inputs) - z_dim = input_z.shape.as_list()[1] vocab_size = input_y.shape.as_list()[1] sess = tf.compat.v1.Session() @@ -57,11 +56,11 @@ def cli(ctx, opt_fp_in, opt_dims, opt_video): else: paths = [opt_fp_in] - fp_inverses = os.path.join(app_cfg.INVERSES_DIR, params.dataset_out) + fp_inverses = os.path.join(app_cfg.INVERSES_DIR, 'inverse_' + int(time.time() * 1000)) os.makedirs(fp_inverses, exist_ok=True) out_file = h5py.File(fp_inverses, 'w') - out_images = out_file.create_dataset('xtrain', [len(paths), 3, 512, 512], dtype='float32') - out_labels = out_file.create_dataset('ytrain', [len(paths), vocab_size], dtype='float32') + out_images = out_file.create_dataset('xtrain', (len(paths), 3, 512, 512,), dtype='float32') + out_labels = out_file.create_dataset('ytrain', (len(paths), vocab_size,), dtype='float32') for path, index in enumerate(paths): fp_frames = find_nearest_vector(generator, sess, input_z, input_y, input_trunc, output, path, opt_dims, out_images, out_labels, index) |