diff options
Diffstat (limited to 'cli/app/search/search_dense.py')
| -rw-r--r-- | cli/app/search/search_dense.py | 7 |
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)]) |