Refactor tasnif.py: Organize imports and update class attributes

cobanov · Mar 11, 2024 · 1d2cf25 · 1d2cf25
1 parent e59e561
commit 1d2cf25
Showing 1 changed file with 9 additions and 3 deletions.
diff --git a/tasnif/tasnif.py b/tasnif/tasnif.py
@@ -6,8 +6,12 @@
 from tqdm import tqdm
 
 from .calculations import calculate_kmeans, calculate_pca, get_embeddings
-from .utils import (create_dir, create_image_grid, read_images_from_directory,
-                    read_with_pil)
+from .utils import (
+    create_dir,
+    create_image_grid,
+    read_images_from_directory,
+    read_with_pil,
+)
 
 warnings.filterwarnings("ignore")
 
@@ -19,10 +23,12 @@ def __init__(self, num_classes, pca_dim=16, use_gpu=False):
         self.use_gpu = use_gpu
         self.embeddings = None
         self.pca_embeddings = None
-        self.centroids = None
+        self.centroid = None
         self.labels = None
         self.image_paths = None
         self.images = []
+        self.project_name = None
+        self.counts = None
 
     def read(self, folder_path):
         """