summaryrefslogtreecommitdiff
path: root/cli/app/search/search_dense.py
diff options
context:
space:
mode:
authorJules Laplace <julescarbon@gmail.com>2020-01-07 22:05:39 +0100
committerJules Laplace <julescarbon@gmail.com>2020-01-07 22:05:39 +0100
commitc6e1c60716a3e4beeaff3987303fd5ae4a30be31 (patch)
treeeb94e6f3b7aef378320591ebf01dfd7244ad111c /cli/app/search/search_dense.py
parent08fa371f49042a2989ec3d494f37b8db63b13c95 (diff)
add stochastic clipping, latent normalization, feature loss to initial pass
Diffstat (limited to 'cli/app/search/search_dense.py')
-rw-r--r--cli/app/search/search_dense.py7
1 files changed, 0 insertions, 7 deletions
diff --git a/cli/app/search/search_dense.py b/cli/app/search/search_dense.py
index a538d0d..8ba3340 100644
--- a/cli/app/search/search_dense.py
+++ b/cli/app/search/search_dense.py
@@ -80,16 +80,9 @@ def find_dense_embedding_for_images(params):
os.makedirs(SAMPLES_DIR, exist_ok=True)
os.makedirs(VECTOR_DIR, exist_ok=True)
- # --------------------------
- # Util functions.
- # --------------------------
- # One hot encoding for classes.
def one_hot(values):
return np.eye(N_CLASS)[values]
- # --------------------------
- # Logging.
- # --------------------------
summary_writer = tf.summary.FileWriter(LOGS_DIR)
def log_stats(name, val, it):
summary = tf.Summary(value=[tf.Summary.Value(tag=name, simple_value=val)])