lazylabel-gui 1.1.8__tar.gz → 1.2.0__tar.gz

{lazylabel_gui-1.1.8/src/lazylabel_gui.egg-info → lazylabel_gui-1.2.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: lazylabel-gui
-Version: 1.1.8
+Version: 1.2.0
 Summary: An image segmentation GUI for generating ML ready mask tensors and annotations.
 Author-email: "Deniz N. Cakan" <deniz.n.cakan@gmail.com>
 License: MIT License
@@ -57,7 +57,7 @@ Dynamic: license-file
 
 **AI-Assisted Image Segmentation Made Simple**
 
-LazyLabel combines Meta's Segment Anything Model (SAM) with intuitive editing tools for fast, precise image labeling. Perfect for machine learning datasets, computer vision research, and annotation workflows.
+LazyLabel combines Meta's Segment Anything Model (SAM) with intuitive editing tools for fast, precise image labeling. Perfect for machine learning datasets and computer vision research.
 
 ![LazyLabel Screenshot](https://raw.githubusercontent.com/dnzckn/LazyLabel/main/src/lazylabel/demo_pictures/gui.PNG)
 
@@ -71,67 +71,57 @@ pip install lazylabel-gui
 lazylabel-gui
 ```
 
+### Optional: SAM-2 Support
+For advanced SAM-2 models, install manually:
+```bash
+pip install git+https://github.com/facebookresearch/sam2.git
+```
+*Note: SAM-2 is optional - LazyLabel works with SAM 1.0 models by default*
+
 ### Usage
 1. **Open Folder** → Select your image directory
 2. **Click on image** → AI generates instant masks  
-3. **Fine-tune** → Edit polygons, merge segments, adjust classes
+3. **Fine-tune** → Edit polygons, merge segments
 4. **Export** → Clean `.npz` files ready for ML training
 
 ---
 
 ## ✨ Key Features
 
-### **🧠 AI-Powered Segmentation**
-- **One-click masking** with Meta's SAM model
-- **Smart prompting** via positive/negative points
-- **Fragment filtering** to remove small artifacts
-- **Multiple model support** (VIT-H, VIT-L, VIT-B)
-
-### **🎨 Advanced Editing**
-- **Polygon drawing** with full vertex control
-- **Bounding box** annotation mode
-- **Shape merging** and class assignment
-- **Edit mode** for precision adjustments
-
-### **⚡ Productivity Tools**
-- **Image adjustments** (brightness, contrast, gamma) 
-- **Customizable hotkeys** for all functions
-- **Undo/redo** with full history
-- **Auto-save** and session persistence
-
-### **📊 ML-Ready Outputs**
-- **One-hot encoded** `.npz` format
-- **Clean class separation** with shape `(H, W, Classes)`
-- **Batch processing** support
-- **Existing mask loading** for iterative work
+- **🧠 One-click AI segmentation** with Meta's SAM and SAM2 models
+- **🎨 Manual polygon drawing** with full vertex control
+- **⚡ Smart editing tools** - merge segments, adjust class names, and class order
+- **📊 ML-ready exports** - One-hot encoded `.npz` format and `.json` for YOLO format
+- **🔧 Image enhancement** - brightness, contrast, gamma adjustment
+- **🔍 Image viewer** - zoom, pan, brightness, contrast, and gamma adjustment
+- **✂️ Edge cropping** - define custom crop areas to focus on specific regions
+- **🔄 Undo/Redo** - full history of all actions
+- **💾 Auto-saving** - Automatic saving of your labels when navigating between images
+- **🎛️ Advanced filtering** - FFT thresholding and color channel thresholding
+- **⌨️ Customizable hotkeys** for all functions
 
 ---
 
-## ⌨️ Essential Controls
-
-| Mode | Key | Action |
-|------|-----|--------|
-| **AI Segmentation** | `1` | Point mode for SAM |
-| | `Left Click` | Add positive point |
-| | `Right Click` | Add negative point |
-| | `Space` | Save segment |
-| **Manual Drawing** | `2` | Polygon mode |
-| | `Left Click` | Add vertex |
-| | `Enter` | Close polygon |
-| **Editing** | `E` | Selection mode |
-| | `R` | Edit selected shapes |
-| | `M` | Merge selected segments |
-| **Navigation** | `Q` | Pan mode |
-| | `W/A/S/D` | Pan image |
-| | `Scroll` | Zoom in/out |
-
-**💡 All hotkeys are customizable** - Click "Hotkeys" button to personalize shortcuts
+## ⌨️ Essential Hotkeys
+
+| Action | Key | Description |
+|--------|-----|-------------|
+| **AI Mode** | `1` | Point-click segmentation |
+| **Draw Mode** | `2` | Manual polygon drawing |
+| **Edit Mode** | `E` | Select and modify shapes |
+| **Save Segment** | `Space` | Confirm current mask |
+| **Merge** | `M` | Combine selected segments |
+| **Pan** | `Q` + drag | Navigate large images |
+| **Positive Point** | `Left Click` | Add to segment |
+| **Negative Point** | `Right Click` | Remove from segment |
+
+💡 **All hotkeys customizable** - Click "Hotkeys" button to personalize
 
 ---
 
 ## 📦 Output Format
 
-LazyLabel exports clean, ML-ready data:
+Perfect for ML training - clean, structured data:
 
 ```python
 import numpy as np
@@ -141,41 +131,24 @@ data = np.load('your_image.npz')
 mask = data['mask']  # Shape: (height, width, num_classes)
 
 # Each channel is a binary mask for one class
-class_0_mask = mask[:, :, 0]  # Binary mask for class 0
-class_1_mask = mask[:, :, 1]  # Binary mask for class 1
-# ... and so on
+class_0_mask = mask[:, :, 0]  # Background
+class_1_mask = mask[:, :, 1]  # Object type 1
+class_2_mask = mask[:, :, 2]  # Object type 2
 ```
 
-**Perfect for:**
-- Semantic segmentation training
-- Instance segmentation datasets  
+
+**Ideal for:**
+- Semantic segmentation datasets
+- Instance segmentation training
 - Computer vision research
 - Automated annotation pipelines
 
 ---
 
-## 🛠️ Advanced Features
-
-### **Image Enhancement**
-- **Brightness/Contrast** adjustment sliders
-- **Gamma correction** for better visibility
-- **Live preview** of adjustments
-- **SAM integration** with adjusted images
-
-### **Smart Filtering**
-- **Fragment threshold** removes small segments
-- **Size-based filtering** (0-100% of largest segment)
-- **Quality control** for clean annotations
-
-### **Professional Workflow**
-- **Class management** with custom aliases
-- **Segment organization** with sortable tables
-- **Batch export** for large datasets
-- **Model switching** without restart
+## 🛠️ Development
 
----
-
-## 🏗️ Development
+**Requirements:** Python 3.10+
+**2.5GB** disk space for SAM model (auto-downloaded)
 
 ### Installation from Source
 ```bash
@@ -185,38 +158,26 @@ pip install -e .
 lazylabel-gui
 ```
 
-### Code Quality & Testing
+### Testing & Quality
 ```bash
-# Linting & formatting
-ruff check . && ruff format .
-
-# Run tests with coverage
+# Run full test suite
 python -m pytest --cov=lazylabel --cov-report=html
 
-# All tests pass with 60%+ coverage
+# Code formatting & linting
+ruff check . && ruff format .
 ```
 
 ### Architecture
-- **Modular design** with clean separation of concerns
-- **Signal-based communication** between components  
+- **Modular design** with clean component separation
+- **Signal-based communication** between UI elements
 - **Extensible model system** for new SAM variants
-- **Comprehensive test suite** with 95% speed optimization
-
----
-
-## 📋 Requirements
-
-- **Python 3.10+**
-- **OpenCV** for image processing
-- **PyQt6** for GUI
-- **NumPy** for data handling
-- **2.5GB** disk space for SAM model (auto-downloaded)
+- **Comprehensive test suite** (150+ tests, 60%+ coverage)
 
 ---
 
 ## 🤝 Contributing
 
-LazyLabel welcomes contributions! Check out our:
+LazyLabel welcomes contributions! Check out:
 - [Architecture Guide](src/lazylabel/ARCHITECTURE.md) for technical details
 - [Hotkey System](src/lazylabel/HOTKEY_FEATURE.md) for customization
 - Issues page for feature requests and bug reports
@@ -225,9 +186,8 @@ LazyLabel welcomes contributions! Check out our:
 
 ## 🙏 Acknowledgments
 
-LazyLabel was inspired by and builds upon the excellent work of:
-- [LabelMe](https://github.com/wkentaro/labelme) - The pioneering open-source image annotation tool
-- [Segment-Anything-UI](https://github.com/branislavhesko/segment-anything-ui) - Early SAM integration concepts
+- [LabelMe](https://github.com/wkentaro/labelme)
+- [Segment-Anything-UI](https://github.com/branislavhesko/segment-anything-ui)
 
 ---
 

lazylabel_gui-1.2.0/README.md

@@ -0,0 +1,145 @@
+# <img src="https://raw.githubusercontent.com/dnzckn/LazyLabel/main/src/lazylabel/demo_pictures/logo2.png" alt="LazyLabel Logo" style="height:60px; vertical-align:middle;" /> <img src="https://raw.githubusercontent.com/dnzckn/LazyLabel/main/src/lazylabel/demo_pictures/logo_black.png" alt="LazyLabel Cursive" style="height:60px; vertical-align:middle;" />
+
+**AI-Assisted Image Segmentation Made Simple**
+
+LazyLabel combines Meta's Segment Anything Model (SAM) with intuitive editing tools for fast, precise image labeling. Perfect for machine learning datasets and computer vision research.
+
+![LazyLabel Screenshot](https://raw.githubusercontent.com/dnzckn/LazyLabel/main/src/lazylabel/demo_pictures/gui.PNG)
+
+---
+
+## 🚀 Quick Start
+
+### Installation
+```bash
+pip install lazylabel-gui
+lazylabel-gui
+```
+
+### Optional: SAM-2 Support
+For advanced SAM-2 models, install manually:
+```bash
+pip install git+https://github.com/facebookresearch/sam2.git
+```
+*Note: SAM-2 is optional - LazyLabel works with SAM 1.0 models by default*
+
+### Usage
+1. **Open Folder** → Select your image directory
+2. **Click on image** → AI generates instant masks  
+3. **Fine-tune** → Edit polygons, merge segments
+4. **Export** → Clean `.npz` files ready for ML training
+
+---
+
+## ✨ Key Features
+
+- **🧠 One-click AI segmentation** with Meta's SAM and SAM2 models
+- **🎨 Manual polygon drawing** with full vertex control
+- **⚡ Smart editing tools** - merge segments, adjust class names, and class order
+- **📊 ML-ready exports** - One-hot encoded `.npz` format and `.json` for YOLO format
+- **🔧 Image enhancement** - brightness, contrast, gamma adjustment
+- **🔍 Image viewer** - zoom, pan, brightness, contrast, and gamma adjustment
+- **✂️ Edge cropping** - define custom crop areas to focus on specific regions
+- **🔄 Undo/Redo** - full history of all actions
+- **💾 Auto-saving** - Automatic saving of your labels when navigating between images
+- **🎛️ Advanced filtering** - FFT thresholding and color channel thresholding
+- **⌨️ Customizable hotkeys** for all functions
+
+---
+
+## ⌨️ Essential Hotkeys
+
+| Action | Key | Description |
+|--------|-----|-------------|
+| **AI Mode** | `1` | Point-click segmentation |
+| **Draw Mode** | `2` | Manual polygon drawing |
+| **Edit Mode** | `E` | Select and modify shapes |
+| **Save Segment** | `Space` | Confirm current mask |
+| **Merge** | `M` | Combine selected segments |
+| **Pan** | `Q` + drag | Navigate large images |
+| **Positive Point** | `Left Click` | Add to segment |
+| **Negative Point** | `Right Click` | Remove from segment |
+
+💡 **All hotkeys customizable** - Click "Hotkeys" button to personalize
+
+---
+
+## 📦 Output Format
+
+Perfect for ML training - clean, structured data:
+
+```python
+import numpy as np
+
+# Load your labeled data
+data = np.load('your_image.npz')
+mask = data['mask']  # Shape: (height, width, num_classes)
+
+# Each channel is a binary mask for one class
+class_0_mask = mask[:, :, 0]  # Background
+class_1_mask = mask[:, :, 1]  # Object type 1
+class_2_mask = mask[:, :, 2]  # Object type 2
+```
+
+
+**Ideal for:**
+- Semantic segmentation datasets
+- Instance segmentation training
+- Computer vision research
+- Automated annotation pipelines
+
+---
+
+## 🛠️ Development
+
+**Requirements:** Python 3.10+
+**2.5GB** disk space for SAM model (auto-downloaded)
+
+### Installation from Source
+```bash
+git clone https://github.com/dnzckn/LazyLabel.git
+cd LazyLabel
+pip install -e .
+lazylabel-gui
+```
+
+### Testing & Quality
+```bash
+# Run full test suite
+python -m pytest --cov=lazylabel --cov-report=html
+
+# Code formatting & linting
+ruff check . && ruff format .
+```
+
+### Architecture
+- **Modular design** with clean component separation
+- **Signal-based communication** between UI elements
+- **Extensible model system** for new SAM variants
+- **Comprehensive test suite** (150+ tests, 60%+ coverage)
+
+---
+
+## 🤝 Contributing
+
+LazyLabel welcomes contributions! Check out:
+- [Architecture Guide](src/lazylabel/ARCHITECTURE.md) for technical details
+- [Hotkey System](src/lazylabel/HOTKEY_FEATURE.md) for customization
+- Issues page for feature requests and bug reports
+
+---
+
+## 🙏 Acknowledgments
+
+- [LabelMe](https://github.com/wkentaro/labelme)
+- [Segment-Anything-UI](https://github.com/branislavhesko/segment-anything-ui)
+
+---
+
+## ☕ Support
+
+If LazyLabel saves you time on annotation tasks, [consider supporting the project!](https://buymeacoffee.com/dnzckn)
+
+---
+
+**Made with ❤️ for the computer vision community**

{lazylabel_gui-1.1.8 → lazylabel_gui-1.2.0}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "lazylabel-gui"
-version = "1.1.8"
+version = "1.2.0"
 authors = [
   { name="Deniz N. Cakan", email="deniz.n.cakan@gmail.com" },
 ]

{lazylabel_gui-1.1.8 → lazylabel_gui-1.2.0}/src/lazylabel/config/hotkeys.py

@@ -44,7 +44,7 @@ class HotkeyManager:
             ),
             HotkeyAction("fit_view", "Fit View", ".", category="Navigation"),
             # Modes
-            HotkeyAction("sam_mode", "Point Mode (SAM)", "1", category="Modes"),
+            HotkeyAction("sam_mode", "AI Mode (Points + Box)", "1", category="Modes"),
             HotkeyAction("polygon_mode", "Polygon Mode", "2", category="Modes"),
             HotkeyAction("bbox_mode", "Bounding Box Mode", "3", category="Modes"),
             HotkeyAction("selection_mode", "Selection Mode", "E", category="Modes"),
@@ -95,7 +95,7 @@ class HotkeyManager:
             # Mouse-related (cannot be reassigned)
             HotkeyAction(
                 "left_click",
-                "Add Positive Point / Select",
+                "AI: Point (click) / Box (drag) / Select",
                 "Left Click",
                 category="Mouse",
                 mouse_related=True,

lazylabel_gui-1.2.0/src/lazylabel/core/model_manager.py

@@ -0,0 +1,183 @@
+"""Model management functionality."""
+
+import os
+from collections.abc import Callable
+
+from ..config import Paths
+from ..models.sam_model import SamModel
+from ..utils.logger import logger
+
+# Optional SAM-2 support
+try:
+    from ..models.sam2_model import Sam2Model
+
+    SAM2_AVAILABLE = True
+except ImportError:
+    logger.info(
+        "SAM-2 not available. Install with: pip install git+https://github.com/facebookresearch/sam2.git"
+    )
+    Sam2Model = None
+    SAM2_AVAILABLE = False
+
+
+class ModelManager:
+    """Manages SAM model loading and selection."""
+
+    def __init__(self, paths: Paths):
+        self.paths = paths
+        self.sam_model: SamModel | None = None
+        self.current_models_folder: str | None = None
+        self.on_model_changed: Callable[[str], None] | None = None
+
+    def initialize_default_model(self, model_type: str = "vit_h") -> SamModel | None:
+        """Initialize the default SAM model.
+
+        Returns:
+            SamModel instance if successful, None if failed
+        """
+        try:
+            logger.info(f"Step 4/8: Loading {model_type.upper()} model...")
+            self.sam_model = SamModel(model_type=model_type)
+            self.current_models_folder = str(self.paths.models_dir)
+            return self.sam_model
+        except Exception as e:
+            logger.error(f"Step 4/8: Failed to initialize default model: {e}")
+            self.sam_model = None
+            return None
+
+    def get_available_models(self, folder_path: str) -> list[tuple[str, str]]:
+        """Get list of available .pth models in folder.
+
+        Returns:
+            List of (display_name, full_path) tuples
+        """
+        pth_files = []
+        for root, _dirs, files in os.walk(folder_path):
+            for file in files:
+                if file.lower().endswith(".pth") or file.lower().endswith(".pt"):
+                    full_path = os.path.join(root, file)
+                    rel_path = os.path.relpath(full_path, folder_path)
+                    pth_files.append((rel_path, full_path))
+
+        return sorted(pth_files, key=lambda x: x[0])
+
+    def detect_model_type(self, model_path: str) -> str:
+        """Detect model type from filename."""
+        filename = os.path.basename(model_path).lower()
+
+        # Check if it's a SAM2 model
+        if self._is_sam2_model(model_path):
+            if "tiny" in filename or "_t" in filename:
+                return "sam2_tiny"
+            elif "small" in filename or "_s" in filename:
+                return "sam2_small"
+            elif "base_plus" in filename or "_b+" in filename:
+                return "sam2_base_plus"
+            elif "large" in filename or "_l" in filename:
+                return "sam2_large"
+            else:
+                return "sam2_large"  # default for SAM2
+        else:
+            # Original SAM model types
+            if "vit_l" in filename or "large" in filename:
+                return "vit_l"
+            elif "vit_b" in filename or "base" in filename:
+                return "vit_b"
+            elif "vit_h" in filename or "huge" in filename:
+                return "vit_h"
+            return "vit_h"  # default for SAM1
+
+    def _is_sam2_model(self, model_path: str) -> bool:
+        """Check if the model is a SAM2 model based on filename patterns."""
+        filename = os.path.basename(model_path).lower()
+        sam2_indicators = ["sam2", "sam2.1", "hiera", "_t.", "_s.", "_b+.", "_l."]
+        return any(indicator in filename for indicator in sam2_indicators)
+
+    def load_custom_model(self, model_path: str) -> bool:
+        """Load a custom model from path.
+
+        Returns:
+            True if successful, False otherwise
+        """
+        if not os.path.exists(model_path):
+            return False
+
+        model_type = self.detect_model_type(model_path)
+
+        try:
+            # Clear existing model from memory
+            if self.sam_model is not None:
+                del self.sam_model
+                import torch
+
+                torch.cuda.empty_cache() if torch.cuda.is_available() else None
+
+            # Create appropriate model instance
+            if self._is_sam2_model(model_path):
+                if not SAM2_AVAILABLE:
+                    logger.warning(
+                        f"SAM-2 model detected but SAM-2 not installed: {model_path}"
+                    )
+                    logger.info(
+                        "Install SAM-2 with: pip install git+https://github.com/facebookresearch/sam2.git"
+                    )
+                    return False
+
+                logger.info(f"Loading SAM2 model: {model_type}")
+                self.sam_model = Sam2Model(model_path)
+            else:
+                logger.info(f"Loading SAM1 model: {model_type}")
+                # Convert SAM2 model types back to SAM1 types for compatibility
+                sam1_model_type = model_type
+                if model_type.startswith("sam2_"):
+                    type_mapping = {
+                        "sam2_tiny": "vit_b",
+                        "sam2_small": "vit_b",
+                        "sam2_base_plus": "vit_l",
+                        "sam2_large": "vit_h",
+                    }
+                    sam1_model_type = type_mapping.get(model_type, "vit_h")
+
+                # Create SAM1 model with custom path
+                self.sam_model = SamModel(
+                    model_type=sam1_model_type, custom_model_path=model_path
+                )
+
+            success = self.sam_model.is_loaded
+
+            if success and self.on_model_changed:
+                model_name = os.path.basename(model_path)
+                self.on_model_changed(f"Current: {model_name}")
+
+            return success
+
+        except Exception as e:
+            logger.error(f"Failed to load custom model: {e}")
+            self.sam_model = None
+            return False
+
+    def set_models_folder(self, folder_path: str) -> None:
+        """Set the current models folder."""
+        self.current_models_folder = folder_path
+
+    def get_models_folder(self) -> str | None:
+        """Get the current models folder."""
+        return self.current_models_folder
+
+    def is_model_available(self) -> bool:
+        """Check if a SAM model is loaded and available."""
+        return self.sam_model is not None and getattr(
+            self.sam_model, "is_loaded", False
+        )
+
+    def set_image_from_path(self, image_path: str) -> bool:
+        """Set image for SAM model from file path."""
+        if not self.is_model_available():
+            return False
+        return self.sam_model.set_image_from_path(image_path)
+
+    def set_image_from_array(self, image_array) -> bool:
+        """Set image for SAM model from numpy array."""
+        if not self.is_model_available():
+            return False
+        return self.sam_model.set_image_from_array(image_array)