From 574562a39fb9b08d3bb6e4383e9b04aa3029bad6 Mon Sep 17 00:00:00 2001 From: Jules Laplace Date: Fri, 9 Oct 2020 13:58:07 +0200 Subject: check if files exist before putting them in the processing pipeline --- cli/app/search/search_class.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'cli/app/search/search_class.py') diff --git a/cli/app/search/search_class.py b/cli/app/search/search_class.py index b0c95bb..901ee0c 100644 --- a/cli/app/search/search_class.py +++ b/cli/app/search/search_class.py @@ -71,7 +71,7 @@ def find_nearest_vector_for_images(paths, opt_dims, opt_steps, opt_video, opt_ta out_fns[index] = os.path.basename(path) fp_frames = find_nearest_vector(sess, generator, path, opt_dims, out_images, out_labels, out_latent, opt_steps, index, opt_tag, opt_stochastic_clipping, opt_label_clipping, opt_use_feature_detector, opt_feature_layers, opt_snapshot_interval, opt_clip_interval, opt_video) - if opt_video: + if opt_video and fp_frames is not None: export_video(fp_frames) sess.close() -- cgit v1.2.3-70-g09d2