summaryrefslogtreecommitdiff
path: root/cli
diff options
context:
space:
mode:
authorJules Laplace <julescarbon@gmail.com>2020-01-08 11:11:54 +0100
committerJules Laplace <julescarbon@gmail.com>2020-01-08 11:11:54 +0100
commit9cbc1bc93aff4a832c9ba4264c32a62034df13a8 (patch)
tree81cdcbbd8cb50d764b136f06b1d46c8a2fc77cec /cli
parentda243d0a0e00e8dd9f4ab3d3d5e973230b554347 (diff)
reprocess
Diffstat (limited to 'cli')
-rw-r--r--cli/app/search/search_class.py4
1 files changed, 2 insertions, 2 deletions
diff --git a/cli/app/search/search_class.py b/cli/app/search/search_class.py
index c732801..25bc1fd 100644
--- a/cli/app/search/search_class.py
+++ b/cli/app/search/search_class.py
@@ -66,13 +66,13 @@ def find_nearest_vector_for_images(paths, opt_dims, opt_steps, opt_video, opt_ta
break
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_stochastic_clipping, opt_label_clipping, opt_use_feature_detector, opt_feature_layers, opt_snapshot_interval, opt_clip_interval, opt_video)
if opt_video:
export_video(fp_frames)
sess.close()
def find_nearest_vector(sess, generator, opt_fp_in, 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_stochastic_clipping, opt_label_clipping, opt_use_feature_detector, opt_feature_layers, opt_snapshot_interval, opt_clip_interval, opt_video):
"""
Find the closest latent and class vectors for an image. Store the class vector in an HDF5.
"""