summaryrefslogtreecommitdiff
path: root/cli/app/search/search_class.py
diff options
context:
space:
mode:
authorJules Laplace <julescarbon@gmail.com>2020-10-09 13:58:07 +0200
committerJules Laplace <julescarbon@gmail.com>2020-10-09 13:58:07 +0200
commit574562a39fb9b08d3bb6e4383e9b04aa3029bad6 (patch)
tree22de10827dcadeb428365696e5a8e31e76e31241 /cli/app/search/search_class.py
parent150fc1dab4b8167cbd1f42e44d809358e40e96fa (diff)
check if files exist before putting them in the processing pipelineHEADmaster
Diffstat (limited to 'cli/app/search/search_class.py')
-rw-r--r--cli/app/search/search_class.py2
1 files changed, 1 insertions, 1 deletions
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()