diff options
| author | Jules Laplace <julescarbon@gmail.com> | 2020-01-15 18:24:15 +0100 |
|---|---|---|
| committer | Jules Laplace <julescarbon@gmail.com> | 2020-01-15 18:24:15 +0100 |
| commit | 4e3ee5ff3090f69b55aa41741d77689ad076eee7 (patch) | |
| tree | febd8498a12dcbd43d8bfb8717a83ca91e64247f /cli/app/commands/biggan | |
| parent | 91b093f86daa2aec3dcd5c6dd335789cd7aae14c (diff) | |
params...
Diffstat (limited to 'cli/app/commands/biggan')
| -rw-r--r-- | cli/app/commands/biggan/extract_dense_vectors.py | 6 | ||||
| -rw-r--r-- | cli/app/commands/biggan/search_dense.py | 5 |
2 files changed, 9 insertions, 2 deletions
diff --git a/cli/app/commands/biggan/extract_dense_vectors.py b/cli/app/commands/biggan/extract_dense_vectors.py index 866104d..4e7952a 100644 --- a/cli/app/commands/biggan/extract_dense_vectors.py +++ b/cli/app/commands/biggan/extract_dense_vectors.py @@ -69,4 +69,8 @@ def cli(ctx, opt_folder_id, opt_latent_steps, opt_dense_steps, opt_video, opt_re ) params = params_dense_dict(tag, folder_id=folder['id']) - find_dense_embedding_for_images(params) + find_dense_embedding_for_images(params, + opt_tag=tag, + opt_feature_layers=opt_feature_layers, + opt_save_progress=False) + diff --git a/cli/app/commands/biggan/search_dense.py b/cli/app/commands/biggan/search_dense.py index b743350..b514ab0 100644 --- a/cli/app/commands/biggan/search_dense.py +++ b/cli/app/commands/biggan/search_dense.py @@ -19,4 +19,7 @@ def cli(ctx, opt_fp_in, opt_tag, opt_feature_layers, opt_save_progress): """ params = Params(opt_fp_in) opt_feature_layers = opt_feature_layers.split(',') - find_dense_embedding_for_images(params, opt_tag=opt_tag, opt_feature_layers=opt_feature_layers, opt_save_progress=opt_save_progress) + find_dense_embedding_for_images(params, + opt_tag=opt_tag, + opt_feature_layers=opt_feature_layers, + opt_save_progress=opt_save_progress) |