fix: minor fix to correct debug statement and correct order arg for c…

…luster
mbari-org · Sep 13, 2024 · 645e979 · 645e979
1 parent 216a4fb
commit 645e979
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/sdcat/cluster/cluster.py b/sdcat/cluster/cluster.py
@@ -323,7 +323,7 @@ def cluster_vits(
 
     # Skip the embedding extraction if all the embeddings are cached
     if num_cached != len(images):
-        debug(f'Extracted embeddings from {num_cached} images...')
+        debug(f'Extracted embeddings from {len(images)} images...')
         compute_norm_embedding(model, images, device)
 
     # Fetch the cached embeddings

diff --git a/sdcat/cluster/commands.py b/sdcat/cluster/commands.py
@@ -256,7 +256,7 @@ def is_day(utc_dt):
 
         # Cluster the detections
         df_cluster = cluster_vits(prefix, model, df, save_dir, alpha, cluster_selection_epsilon, cluster_selection_method,
-                                  min_similarity, min_cluster_size, min_samples, use_tsne, device)
+                                  min_similarity, min_cluster_size, min_samples, device,  use_tsne, roi=False)
 
         # Merge the results with the original DataFrame
         df.update(df_cluster)