From 08fa371f49042a2989ec3d494f37b8db63b13c95 Mon Sep 17 00:00:00 2001 From: Jules Laplace Date: Tue, 7 Jan 2020 20:05:57 +0100 Subject: script to run everything --- cli/app/commands/biggan/search_class.py | 9 ++++++++- 1 file changed, 8 insertions(+), 1 deletion(-) (limited to 'cli/app/commands/biggan/search_class.py') diff --git a/cli/app/commands/biggan/search_class.py b/cli/app/commands/biggan/search_class.py index 0501729..311dc70 100644 --- a/cli/app/commands/biggan/search_class.py +++ b/cli/app/commands/biggan/search_class.py @@ -21,4 +21,11 @@ def cli(ctx, opt_fp_in, opt_dims, opt_steps, opt_limit, opt_video, opt_tag): """ Search for an image (class vector) in BigGAN using gradient descent """ - find_nearest_vector_for_images(opt_fp_in, opt_dims, opt_steps, opt_limit, opt_video, opt_tag) + if os.path.isdir(opt_fp_in): + 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')) + else: + paths = [opt_fp_in] + + find_nearest_vector_for_images(paths, opt_dims, opt_steps, opt_video, opt_tag, opt_limit) -- cgit v1.2.3-70-g09d2