diff --git a/sdcat/cluster/cluster.py b/sdcat/cluster/cluster.py index fd60987..bba5aea 100755 --- 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 index 9a1fd63..3835c33 100644 --- 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)