active-vision 0.4.2__py3-none-any.whl → 0.4.3__py3-none-any.whl

active_vision/__init__.py

@@ -1,3 +1,3 @@
-__version__ = "0.4.2"
+__version__ = "0.4.3"
 
 from .core import *

active_vision/core.py

@@ -308,6 +308,7 @@ class ActiveLearner:
                 "label": self.eval_set[label_col].tolist(),
                 "pred_label": [self.learn.dls.vocab[i] for i in cls_preds.numpy()],
                 "pred_conf": torch.max(F.softmax(probs, dim=1), dim=1)[0].numpy(),
+                "probs": probs.numpy().tolist(),
                 "loss": loss.numpy().tolist(),
             }
         )
@@ -324,9 +325,17 @@ class ActiveLearner:
                 get_base64_image
             )
             interactive_eval_df = interactive_eval_df[
-                ["image", "filepath", "label", "pred_label", "pred_conf", "loss"]
+                [
+                    "image",
+                    "filepath",
+                    "label",
+                    "pred_label",
+                    "pred_conf",
+                    "loss",
+                    "probs",
+                ]
             ]
-            
+
             show(
                 interactive_eval_df,
                 columnDefs=[{"width": "200px", "targets": "_all"}],
@@ -337,7 +346,11 @@ class ActiveLearner:
         return eval_df
 
     def sample_uncertain(
-        self, df: pd.DataFrame, num_samples: int, strategy: str = "least-confidence", interactive: bool = False
+        self,
+        df: pd.DataFrame,
+        num_samples: int,
+        strategy: str = "least-confidence",
+        interactive: bool = False,
     ):
         """
         Sample top `num_samples` low confidence samples. Returns a df with filepaths and predicted labels, and confidence scores.
@@ -442,7 +455,11 @@ class ActiveLearner:
         return result_df
 
     def sample_diverse(
-        self, df: pd.DataFrame, num_samples: int, strategy: str = "model-based-outlier", interactive: bool = False
+        self,
+        df: pd.DataFrame,
+        num_samples: int,
+        strategy: str = "model-based-outlier",
+        interactive: bool = False,
     ):
         """
         Sample top `num_samples` diverse samples. Returns a df with filepaths and predicted labels, and confidence scores.
@@ -526,9 +543,18 @@ class ActiveLearner:
             if interactive:
                 logger.info("Rendering interactive table")
                 interactive_df = result_df.copy()
-                interactive_df["image"] = interactive_df["filepath"].apply(get_base64_image)
+                interactive_df["image"] = interactive_df["filepath"].apply(
+                    get_base64_image
+                )
                 interactive_df = interactive_df[
-                    ["image", "filepath", "strategy", "score", "pred_label", "pred_conf"]
+                    [
+                        "image",
+                        "filepath",
+                        "strategy",
+                        "score",
+                        "pred_label",
+                        "pred_conf",
+                    ]
                 ]
 
                 show(
@@ -544,7 +570,13 @@ class ActiveLearner:
             logger.error(f"Unknown strategy: {strategy}")
             raise ValueError(f"Unknown strategy: {strategy}")
 
-    def sample_random(self, df: pd.DataFrame, num_samples: int, seed: int = None, interactive: bool = False):
+    def sample_random(
+        self,
+        df: pd.DataFrame,
+        num_samples: int,
+        seed: int = None,
+        interactive: bool = False,
+    ):
         """
         Sample `num_samples` random samples. Returns a df with filepaths and predicted labels, and confidence scores.
 
@@ -580,7 +612,13 @@ class ActiveLearner:
 
         return result_df
 
-    def sample_combination(self, df: pd.DataFrame, num_samples: int, combination: dict, interactive: bool = False):
+    def sample_combination(
+        self,
+        df: pd.DataFrame,
+        num_samples: int,
+        combination: dict,
+        interactive: bool = False,
+    ):
         """
         Sample samples based on a combination of strategies.
 
@@ -614,7 +652,9 @@ class ActiveLearner:
 
         # Validate total proportions sum to 1
         if not np.isclose(sum(combination.values()), 1.0):
-            raise ValueError(f"Proportions must sum to 1, got {sum(combination.values())}")
+            raise ValueError(
+                f"Proportions must sum to 1, got {sum(combination.values())}"
+            )
 
         # Calculate samples per strategy and handle rounding
         samples_per_strategy = {
@@ -648,7 +688,9 @@ class ActiveLearner:
                     df=df, num_samples=n_samples, strategy=strategy, interactive=False
                 )
             elif strategy == "random":
-                strategy_df = self.sample_random(df=df, num_samples=n_samples, interactive=False)
+                strategy_df = self.sample_random(
+                    df=df, num_samples=n_samples, interactive=False
+                )
             else:
                 raise ValueError(f"Unknown strategy: {strategy}")
 
@@ -852,7 +894,9 @@ class ActiveLearner:
                     )
 
                     # Add element_id to finish button
-                    finish_btn = gr.Button("Finish Labeling", variant="primary", elem_id="finish_btn")
+                    finish_btn = gr.Button(
+                        "Finish Labeling", variant="primary", elem_id="finish_btn"
+                    )
 
                     # Add event handler for slider changes
                     progress.change(

active_vision/utils.py

@@ -1,6 +1,8 @@
 import base64
+import os
 from io import BytesIO
 
+from itables import show
 from loguru import logger
 from PIL import Image
 
@@ -11,7 +13,7 @@ def get_base64_image(filepath, width=200):
             # Convert to RGB if needed
             if img.mode != "RGB":
                 img = img.convert("RGB")
-            
+
             aspect_ratio = img.height / img.width
             height = int(width * aspect_ratio)
             img = img.resize((width, height), Image.Resampling.LANCZOS)
@@ -22,3 +24,35 @@ def get_base64_image(filepath, width=200):
     except Exception as e:
         logger.warning(f"Failed to encode image {filepath}: {e}")
         return None
+
+
+def show_interactive_table(df, filepath_col="filepath", image_col="image"):
+    """
+    Display an interactive table with images from filepaths.
+
+    Args:
+        df: pandas DataFrame that contains a 'filepath' column
+    """
+    # Create a copy to avoid modifying the original dataframe
+    interactive_df = df.copy()
+
+    # Add image column by applying get_base64_image to filepath column
+    interactive_df[image_col] = interactive_df[filepath_col].apply(get_base64_image)
+
+    # Convert filepath to clickable link with relative path
+    interactive_df[filepath_col] = interactive_df[filepath_col].apply(
+        lambda x: f'<a href="{x.replace(os.sep, "/")}">{x}</a>'
+    )
+
+    # Reorder columns to show image first
+    cols = interactive_df.columns.tolist()
+    cols.remove(image_col)
+    interactive_df = interactive_df[[image_col] + cols]
+
+    # Display interactive table
+    show(
+        interactive_df,
+        columnDefs=[{"width": "200px", "targets": "_all"}],
+        style="width:1200px",
+        autoWidth=False,
+    )

{active_vision-0.4.2.dist-info → active_vision-0.4.3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: active-vision
-Version: 0.4.2
+Version: 0.4.3
 Summary: Active learning for computer vision.
 Project-URL: Homepage, https://github.com/dnth/active-vision
 Project-URL: Bug Tracker, https://github.com/dnth/active-vision/issues
@@ -156,7 +156,7 @@ al = ActiveLearner(name="cycle-1")
 al.load_model(model="resnet18", pretrained=True)
 
 # Load dataset 
-al.load_dataset(train_df, filepath_col="filepath", label_col="label", batch_size=8)
+al.load_dataset(train_df, filepath_col="filepath", label_col="label")
 
 # Train model
 al.train(epochs=10, lr=5e-3)

active_vision-0.4.3.dist-info/RECORD

@@ -0,0 +1,7 @@
+active_vision/__init__.py,sha256=ztL09ANIuHmKmue2Uaui475201zOvb4sEYHNspIzDEA,43
+active_vision/core.py,sha256=9Qwmrere_ryQKqFKLm0aRp-3drz2yBzHfm6jumjWAto,47765
+active_vision/utils.py,sha256=L2nIUSohqCABqV8qjKNTHmxQoiom2DXMjce5aw2_Tjc,1902
+active_vision-0.4.3.dist-info/METADATA,sha256=JLNqzChj2bhefMu9a4HSrimzImsYm1ePuJhJbTNUWe8,19798
+active_vision-0.4.3.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+active_vision-0.4.3.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+active_vision-0.4.3.dist-info/RECORD,,

active_vision-0.4.2.dist-info/RECORD

@@ -1,7 +0,0 @@
-active_vision/__init__.py,sha256=WHlXpfYbX7Qp1P6kfuEm60U1xl8_A6fQPVJDSLjojYo,43
-active_vision/core.py,sha256=51fskr1E2QLwgxuKzv75yxqH73ktIxEPovxqN7LWibM,47081
-active_vision/utils.py,sha256=IV8RbxCX2Owk0-fF0-LK53YJr8erHQjYb6gSrj9oACo,841
-active_vision-0.4.2.dist-info/METADATA,sha256=v_fUTAj-2vn33eQJH3aafPAMw3KINJe-u9XK-2Nojyk,19812
-active_vision-0.4.2.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-active_vision-0.4.2.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-active_vision-0.4.2.dist-info/RECORD,,