diff options
| author | Jules Laplace <julescarbon@gmail.com> | 2019-12-09 09:59:48 +0100 |
|---|---|---|
| committer | Jules Laplace <julescarbon@gmail.com> | 2019-12-09 09:59:48 +0100 |
| commit | 40300c85d0b622e1f23eb7f2c6f57e6df46d476e (patch) | |
| tree | 1d5d67f67417618c7a8441bf5c80adf44ff3eb1e /cli/app/commands | |
| parent | 898c9cf94d9e58b1174463a08caa51400d529311 (diff) | |
fix
Diffstat (limited to 'cli/app/commands')
| -rw-r--r-- | cli/app/commands/biggan/search.py | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/cli/app/commands/biggan/search.py b/cli/app/commands/biggan/search.py index 600e698..b15e202 100644 --- a/cli/app/commands/biggan/search.py +++ b/cli/app/commands/biggan/search.py @@ -114,9 +114,9 @@ def cli(ctx, opt_fp_in, opt_dims): sess.run(tf.compat.v1.tables_initializer()) if os.path.isdir(opt_fp_in): - paths = glob(os.path.join(params.input_dir, '*.jpg')) + \ - glob(os.path.join(params.input_dir, '*.jpeg')) + \ - glob(os.path.join(params.input_dir, '*.png')) + paths = glob(os.path.join(opt_fp_in, '*.jpg')) + \ + glob(os.path.join(opt_fp_in, '*.jpeg')) + \ + glob(os.path.join(opt_fp_in, '*.png')) for path in paths: find_nearest_vector(generator, sess, input_z, input_y, input_trunc, output, path, opt_dims) else: |