summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorJules Laplace <julescarbon@gmail.com>2020-01-08 01:17:08 +0100
committerJules Laplace <julescarbon@gmail.com>2020-01-08 01:17:08 +0100
commit9f0c0691a4f0c1e54d7216e4596df0ea4b8c7226 (patch)
tree22bf26560e2bc29e6d438f85acfbdb074c53843e
parentc00bc938019f660b72fff7f9972f93eb7462d8cf (diff)
getting latent search updates working...
-rw-r--r--cli/app/search/json.py2
-rw-r--r--cli/app/search/search_class.py2
2 files changed, 2 insertions, 2 deletions
diff --git a/cli/app/search/json.py b/cli/app/search/json.py
index b47512d..4403353 100644
--- a/cli/app/search/json.py
+++ b/cli/app/search/json.py
@@ -3,7 +3,7 @@ from app.settings import app_cfg
from app.utils.file_utils import write_json
from app.search.params import ParamsDict
-def make_params_latent(fp_out_dir, tag):
+def make_params_latent(tag):
return {
"tag": tag,
"decay_n": 2,
diff --git a/cli/app/search/search_class.py b/cli/app/search/search_class.py
index 2296cb9..f13e1fd 100644
--- a/cli/app/search/search_class.py
+++ b/cli/app/search/search_class.py
@@ -53,7 +53,7 @@ def find_nearest_vector_for_images(paths, opt_dims, opt_steps, opt_video, opt_ta
fp_inverses = os.path.join(app_cfg.DIR_INVERSES, opt_tag)
os.makedirs(fp_inverses, exist_ok=True)
- save_params_latent(fp_inverses, opt_tag)
+ # save_params_latent(fp_inverses, opt_tag)
save_params_dense(fp_inverses, opt_tag)
out_file = h5py.File(join(fp_inverses, 'dataset.latent.hdf5'), 'w')
out_images = out_file.create_dataset('xtrain', (len(paths), 3, 512, 512,), dtype='float32')