lazylabel-gui 1.0.9__py3-none-any.whl → 1.1.1__py3-none-any.whl

lazylabel/ui/widgets/status_bar.py

@@ -0,0 +1,109 @@
+"""Status bar widget for displaying active messages."""
+
+from PyQt6.QtWidgets import QStatusBar, QLabel
+from PyQt6.QtCore import QTimer, pyqtSignal, Qt
+from PyQt6.QtGui import QFont
+
+
+class StatusBar(QStatusBar):
+    """Custom status bar for displaying messages and app status."""
+
+    def __init__(self, parent=None):
+        super().__init__(parent)
+        self._message_timer = QTimer()
+        self._message_timer.timeout.connect(self._clear_temporary_message)
+        self._setup_ui()
+
+    def _setup_ui(self):
+        """Setup the status bar UI."""
+        # Set a reasonable height for the status bar
+        self.setFixedHeight(25)
+
+        # Main message label (centered)
+        self.message_label = QLabel()
+        self.message_label.setStyleSheet("color: #ffa500; padding: 2px 5px;")
+        self.message_label.setAlignment(Qt.AlignmentFlag.AlignCenter)
+        font = QFont()
+        font.setPointSize(9)
+        self.message_label.setFont(font)
+
+        # Add the message label as the main widget
+        self.addWidget(self.message_label, 1)  # stretch factor 1
+
+        # Permanent status label (right side)
+        self.permanent_label = QLabel()
+        self.permanent_label.setStyleSheet("color: #888; padding: 2px 5px;")
+        font = QFont()
+        font.setPointSize(9)
+        self.permanent_label.setFont(font)
+        self.addPermanentWidget(self.permanent_label)
+
+        # Default state
+        self.set_ready_message()
+
+    def show_message(self, message: str, duration: int = 5000):
+        """Show a temporary message for specified duration."""
+        self.message_label.setText(message)
+        self.message_label.setStyleSheet("color: #ffa500; padding: 2px 5px;")
+
+        # Stop any existing timer
+        self._message_timer.stop()
+
+        # Start new timer if duration > 0
+        if duration > 0:
+            self._message_timer.start(duration)
+
+    def show_error_message(self, message: str, duration: int = 8000):
+        """Show an error message with red color."""
+        self.message_label.setText(f"Error: {message}")
+        self.message_label.setStyleSheet("color: #ff6b6b; padding: 2px 5px;")
+
+        # Stop any existing timer
+        self._message_timer.stop()
+
+        # Start new timer if duration > 0
+        if duration > 0:
+            self._message_timer.start(duration)
+
+    def show_success_message(self, message: str, duration: int = 3000):
+        """Show a success message with green color."""
+        self.message_label.setText(message)
+        self.message_label.setStyleSheet("color: #51cf66; padding: 2px 5px;")
+
+        # Stop any existing timer
+        self._message_timer.stop()
+
+        # Start new timer if duration > 0
+        if duration > 0:
+            self._message_timer.start(duration)
+
+    def show_warning_message(self, message: str, duration: int = 5000):
+        """Show a warning message with yellow color."""
+        self.message_label.setText(f"Warning: {message}")
+        self.message_label.setStyleSheet("color: #ffd43b; padding: 2px 5px;")
+
+        # Stop any existing timer
+        self._message_timer.stop()
+
+        # Start new timer if duration > 0
+        if duration > 0:
+            self._message_timer.start(duration)
+
+    def set_permanent_message(self, message: str):
+        """Set a permanent message (usually for status info)."""
+        self.permanent_label.setText(message)
+
+    def set_ready_message(self):
+        """Set the default ready message."""
+        self.message_label.setText("")  # Blank instead of "Ready"
+        self.message_label.setStyleSheet("color: #888; padding: 2px 5px;")
+        self._message_timer.stop()
+
+    def _clear_temporary_message(self):
+        """Clear temporary message and return to ready state."""
+        self.set_ready_message()
+        self._message_timer.stop()
+
+    def clear_message(self):
+        """Immediately clear any message."""
+        self.set_ready_message()

lazylabel/utils/__init__.py

@@ -0,0 +1,6 @@
+"""Utility modules."""
+
+from .utils import mask_to_pixmap
+from .custom_file_system_model import CustomFileSystemModel
+
+__all__ = ['mask_to_pixmap', 'CustomFileSystemModel']

lazylabel/{custom_file_system_model.py → utils/custom_file_system_model.py}

@@ -39,7 +39,7 @@ class CustomFileSystemModel(QFileSystemModel):
             pass
 
     def update_cache_for_path(self, saved_file_path: str):
-        """Incrementally updates the cache and the view for a newly saved file."""
+        """Incrementally updates the cache and the view for a newly saved or deleted file."""
         if not saved_file_path:
             return
 
@@ -47,9 +47,15 @@ class CustomFileSystemModel(QFileSystemModel):
         base_name = p.stem
 
         if p.suffix == ".npz":
-            self.npz_files.add(base_name)
+            if p.exists():
+                self.npz_files.add(base_name)
+            else:
+                self.npz_files.discard(base_name)
         elif p.suffix == ".txt":
-            self.txt_files.add(base_name)
+            if p.exists():
+                self.txt_files.add(base_name)
+            else:
+                self.txt_files.discard(base_name)
         else:
             return
 

{lazylabel_gui-1.0.9.dist-info → lazylabel_gui-1.1.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: lazylabel-gui
-Version: 1.0.9
+Version: 1.1.1
 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
@@ -48,7 +48,8 @@ Requires-Dist: tqdm>=4.67.1
 Dynamic: license-file
 
 # <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;" />
-LazyLabel is an intuitive, AI-assisted image segmentation tool. It uses Meta's Segment Anything Model (SAM) for quick, precise mask generation, alongside advanced polygon editing for fine-tuned control. Outputs are saved in a clean, one-hot encoded `.npz` format for easy machine learning integration and in YOLO `.txt` format.
+
+LazyLabel is an intuitive, AI-assisted image segmentation tool built with a modern, modular architecture. It uses Meta's Segment Anything Model (SAM) for quick, precise mask generation, alongside advanced polygon editing for fine-tuned control. Features comprehensive model management, customizable hotkeys, and outputs in clean, one-hot encoded `.npz` format for easy machine learning integration.
 
 Inspired by [LabelMe](https://github.com/wkentaro/labelme?tab=readme-ov-file#installation) and [Segment-Anything-UI](https://github.com/branislavhesko/segment-anything-ui/tree/main).
 
@@ -58,19 +59,33 @@ Inspired by [LabelMe](https://github.com/wkentaro/labelme?tab=readme-ov-file#ins
 
 ## ✨ Core Features
 
-* **AI-Powered Segmentation**: Generate masks with simple left-click (positive) and right-click (negative) interactions.
-* **Vector Polygon Tool**: Full control to draw, edit, and reshape polygons. Drag vertices or move entire shapes.
-* **Advanced Class Management**: Assign multiple segments to a single class ID for organized labeling.
-* **Intuitive Editing & Refinement**: Select, merge, and re-order segments.
-* **Interactive UI**: Color-coded segments, sortable lists, and hover highlighting.
-* **Smart I/O**: Loads existing `.npz` masks; saves work as clean, one-hot encoded outputs.
+### **AI-Powered Segmentation**
+* Generate masks with simple left-click (positive) and right-click (negative) interactions
+* Multiple SAM model support with easy switching
+* Custom model loading from any directory
+
+### **Advanced Editing Tools**
+* **Vector Polygon Tool**: Full control to draw, edit, and reshape polygons
+* **Vertex Editing**: Drag vertices or move entire shapes with precision
+* **Selection & Merging**: Select, merge, and re-order segments intuitively
+
+### **Professional Workflow**
+* **Customizable Hotkeys**: Personalize keyboard shortcuts for all functions
+* **Advanced Class Management**: Assign multiple segments to single class IDs
+* **Smart I/O**: Load existing `.npz` masks; save as clean, one-hot encoded outputs
+* **Interactive UI**: Color-coded segments, sortable lists, and hover highlighting
+
+### **Modern Architecture**
+* **Modular Design**: Clean, maintainable codebase with separated concerns
+* **Model Management**: Dedicated model storage and switching system
+* **Persistent Settings**: User preferences saved between sessions
 
 ---
 
 ## 🚀 Getting Started
 
 ### Prerequisites
-**Python 3.10**
+**Python 3.10+**
 
 ### Installation
 
@@ -90,7 +105,7 @@ Inspired by [LabelMe](https://github.com/wkentaro/labelme?tab=readme-ov-file#ins
     git clone https://github.com/dnzckn/LazyLabel.git
     cd LazyLabel
     ```
-2.  Install in editable mode, which links the installed package to your source directory:
+2.  Install in editable mode:
     ```bash
     pip install -e .
     ```
@@ -99,12 +114,20 @@ Inspired by [LabelMe](https://github.com/wkentaro/labelme?tab=readme-ov-file#ins
     lazylabel-gui
     ```
 
-**Note**: On the first run, the application will automatically download the SAM model checkpoint (~2.5 GB) from Meta's repository to a local cache. This is a one-time download.
+### Model Management
+* **Default Storage**: Models are stored in `src/lazylabel/models/` directory
+* **Custom Models**: Click "Browse Models" to select custom model folders  
+* **Model Switching**: Use the dropdown to switch between available models
+* **Auto-Detection**: Application automatically detects all `.pth` files in selected directories
+
+**Note**: On the first run, the application will automatically download the SAM model checkpoint (~2.5 GB) from Meta's repository to the models directory. This is a one-time download.
 
 ---
 
 ## ⌨️ Controls & Keybinds
 
+> **💡 Tip**: All hotkeys are fully customizable! Click the "Hotkeys" button in the control panel to personalize your shortcuts.
+
 ### Modes
 | Key | Action |
 |---|---|
@@ -143,5 +166,32 @@ Each channel is a binary mask for a class, combining all assigned segments into
 
 ---
 
+## 🏗️ Architecture
+
+LazyLabel features a modern, modular architecture designed for maintainability and extensibility:
+
+* **Modular Design**: Clean separation between UI, business logic, and configuration
+* **Signal-Based Communication**: Loose coupling between components using PyQt signals
+* **Persistent Configuration**: User settings and preferences saved between sessions
+* **Extensible Model System**: Easy integration of new SAM models and types
+
+For detailed technical documentation, see [ARCHITECTURE.md](src/lazylabel/ARCHITECTURE.md).
+
+---
+
+## ⌨️ Hotkey Customization
+
+LazyLabel includes a comprehensive hotkey management system:
+
+* **Full Customization**: Personalize keyboard shortcuts for all 27+ functions
+* **Category Organization**: Hotkeys organized by function (Modes, Actions, Navigation, etc.)
+* **Primary & Secondary Keys**: Set multiple shortcuts for the same action
+* **Persistent Settings**: Custom hotkeys saved between sessions
+* **Conflict Prevention**: System prevents duplicate key assignments
+
+For complete hotkey documentation, see [HOTKEY_FEATURE.md](src/lazylabel/HOTKEY_FEATURE.md).
+
+---
+
 ## ☕ Support LazyLabel
 [If you found LazyLabel helpful, consider supporting the project!](https://buymeacoffee.com/dnzckn)

lazylabel_gui-1.1.1.dist-info/RECORD

@@ -0,0 +1,37 @@
+lazylabel/__init__.py,sha256=XTAPk-88MdkJX5IKg0f51CxNlARQ-SZIiqLneV6B3II,197
+lazylabel/main.py,sha256=U0_bYNiVmmwdhRgBOryRZE91sunDQYEJTGGMNFubj1M,766
+lazylabel/config/__init__.py,sha256=ahAF6cneEVmq2zyGwydGgGAOzJKkv_aumq5cWmokOms,261
+lazylabel/config/hotkeys.py,sha256=LrQ4nmLWVOZvNOe4mEARFU3Wd3bZ7nvvsxGeFY6N9wA,7618
+lazylabel/config/paths.py,sha256=d6lJ0rwdlnLjrhR5fIzb3Pc9Fhl4lsGIf2IZNmi4vZ8,1320
+lazylabel/config/settings.py,sha256=IkUGz-FdDapF71k_LRpUOLHScglGeYjdLbbELGJ4hOs,1860
+lazylabel/core/__init__.py,sha256=YIj3IAI6Er5nN7gSuo9E8eoQKbBbSq2Jy2prtU8Fp50,230
+lazylabel/core/file_manager.py,sha256=cXaSPSUiqTm57hfpUxkuzgqApOZCK1RAJdfjmKRJKRA,4734
+lazylabel/core/model_manager.py,sha256=pLTmQlVfambA3y4aypHu_jn_WrW0ZrMu_8j023mr_EM,3435
+lazylabel/core/segment_manager.py,sha256=kLEj5S-fEmoXIiJ2Em97GJlre8OOpSotX5bIWMrBCvE,6006
+lazylabel/models/__init__.py,sha256=qH0EvkWsou0boS85yM6DfRhJnrPOLc-QzWI0grAwnRI,89
+lazylabel/models/sam_model.py,sha256=lMakV3_B-hNJ3BXy1sErkSzH9YvDsColevXLGaeLnP4,7762
+lazylabel/ui/__init__.py,sha256=i5hgblzrydTkFSJDiFyfRZMNl4Z4J2dumuwx3bWEPvA,266
+lazylabel/ui/control_panel.py,sha256=Pfv4D3V9oBkTWWzTE944JAUCa0by7jv9h3EruKnJLVM,8894
+lazylabel/ui/editable_vertex.py,sha256=pMuXvlCkbguUsFisEzj3bC2UZ_xKqfGuM8kyPug5yws,1852
+lazylabel/ui/hotkey_dialog.py,sha256=3QkthYil_SWYWxRGL_U8cZCsQ4y-6SExQZXJuE5aO8s,15252
+lazylabel/ui/hoverable_pixelmap_item.py,sha256=kJFOp7WXiyHpNf7l73TZjiob85jgP30b5MZvu_z5L3c,728
+lazylabel/ui/hoverable_polygon_item.py,sha256=hQKax3vzhtrdB5ThwTINjOCwOYy718zw5YZPLCfLnGM,1251
+lazylabel/ui/main_window.py,sha256=zjpTH-RqBx5mJ1i1Y0CC8KepgQF28mIJDYc68CUoBIk,64488
+lazylabel/ui/numeric_table_widget_item.py,sha256=dQUlIFu9syCxTGAHVIlmbgkI7aJ3f3wmDPBz1AGK9Bg,283
+lazylabel/ui/photo_viewer.py,sha256=PNgm0gU2gnIqvRkrGlQugdobGsKwAi3m3X6ZF487lCo,2055
+lazylabel/ui/reorderable_class_table.py,sha256=4c-iuSkPcmk5Aey5n2zz49O85x9TQPujKG-JLxtuBCo,2406
+lazylabel/ui/right_panel.py,sha256=7ksUM7eItHrdXWkuDHohXEShdYYR2FYx-zeSmbfi620,12454
+lazylabel/ui/widgets/__init__.py,sha256=5wJ09bTPlhqHyBIXGDeXiNJf33YelKKLu0aPrar9AxU,314
+lazylabel/ui/widgets/adjustments_widget.py,sha256=Az8GZgu4Uvhm6nEnwjiYDyShOqwieinjFnTOiUfxdzo,3940
+lazylabel/ui/widgets/model_selection_widget.py,sha256=X3qVH90yCaCpLDauj-DLWJgAwAkAVQrzhG9X5mESa-o,3590
+lazylabel/ui/widgets/settings_widget.py,sha256=8zhjLxUxqFqxqMYDzkXWD1Ye5V-a7HNhPjPJjYBxeZw,4361
+lazylabel/ui/widgets/status_bar.py,sha256=71paNOXlxXmWbCG6luyciXRmqC7DNFYDi_Eo2HDzjOg,4026
+lazylabel/utils/__init__.py,sha256=SX_aZvmFojIJ4Lskat9ly0dmFBXgN7leBdmc68aDLpg,177
+lazylabel/utils/custom_file_system_model.py,sha256=zK0Z4LY2hzYVtycDWsCLEhXZ0zKZz-Qy5XGwHYrqazQ,4867
+lazylabel/utils/utils.py,sha256=sYSCoXL27OaLgOZaUkCAhgmKZ7YfhR3Cc5F8nDIa3Ig,414
+lazylabel_gui-1.1.1.dist-info/licenses/LICENSE,sha256=kSDEIgrWAPd1u2UFGGpC9X71dhzrlzBFs8hbDlENnGE,1092
+lazylabel_gui-1.1.1.dist-info/METADATA,sha256=RMI2kMGM08gRUJqkKXrMNbkKvmNDawmvjRCutT_Y0bM,8473
+lazylabel_gui-1.1.1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+lazylabel_gui-1.1.1.dist-info/entry_points.txt,sha256=Hd0WwEG9OPTa_ziYjiD0aRh7R6Fupt-wdQ3sspdc1mM,54
+lazylabel_gui-1.1.1.dist-info/top_level.txt,sha256=YN4uIyrpDBq1wiJaBuZLDipIzyZY0jqJOmmXiPIOUkU,10
+lazylabel_gui-1.1.1.dist-info/RECORD,,

lazylabel/controls.py

@@ -1,265 +0,0 @@
-from PyQt6.QtWidgets import (
-    QWidget,
-    QVBoxLayout,
-    QPushButton,
-    QLabel,
-    QFrame,
-    QTableWidget,
-    QTreeView,
-    QAbstractItemView,
-    QHBoxLayout,
-    QComboBox,
-    QHeaderView,
-    QCheckBox,
-    QSlider,
-    QGroupBox,
-    QSplitter,
-)
-from PyQt6.QtCore import Qt
-from .reorderable_class_table import ReorderableClassTable
-
-
-class ControlPanel(QWidget):
-    def __init__(self, parent=None):
-        super().__init__(parent)
-        layout = QVBoxLayout(self)
-        layout.setAlignment(Qt.AlignmentFlag.AlignTop)
-
-        toggle_layout = QHBoxLayout()
-        self.btn_toggle_visibility = QPushButton("< Hide")
-        self.btn_toggle_visibility.setToolTip("Hide this panel")
-        toggle_layout.addWidget(self.btn_toggle_visibility)
-        toggle_layout.addStretch()
-        layout.addLayout(toggle_layout)
-
-        self.main_controls_widget = QWidget()
-        main_layout = QVBoxLayout(self.main_controls_widget)
-        main_layout.setContentsMargins(0, 0, 0, 0)
-
-        self.mode_label = QLabel("Mode: Points")
-        font = self.mode_label.font()
-        font.setPointSize(14)
-        font.setBold(True)
-        self.mode_label.setFont(font)
-        main_layout.addWidget(self.mode_label)
-
-        self.btn_sam_mode = QPushButton("Point Mode (1)")
-        self.btn_sam_mode.setToolTip("Switch to Point Mode for AI segmentation (1)")
-        self.btn_polygon_mode = QPushButton("Polygon Mode (2)")
-        self.btn_polygon_mode.setToolTip("Switch to Polygon Drawing Mode (2)")
-        self.btn_selection_mode = QPushButton("Selection Mode (E)")
-        self.btn_selection_mode.setToolTip("Toggle segment selection (E)")
-        main_layout.addWidget(self.btn_sam_mode)
-        main_layout.addWidget(self.btn_polygon_mode)
-        main_layout.addWidget(self.btn_selection_mode)
-
-        main_layout.addSpacing(20)
-        line1 = QFrame()
-        line1.setFrameShape(QFrame.Shape.HLine)
-        main_layout.addWidget(line1)
-        main_layout.addSpacing(10)
-
-        self.btn_fit_view = QPushButton("Fit View (.)")
-        self.btn_fit_view.setToolTip("Reset image zoom and pan to fit the view (.)")
-        self.btn_clear_points = QPushButton("Clear Clicks (C)")
-        self.btn_clear_points.setToolTip("Clear current temporary points/vertices (C)")
-        main_layout.addWidget(self.btn_fit_view)
-        main_layout.addWidget(self.btn_clear_points)
-
-        main_layout.addSpacing(10)
-
-        settings_group = QGroupBox("Settings")
-        settings_layout = QVBoxLayout()
-
-        self.chk_auto_save = QCheckBox("Auto-Save on Navigate")
-        self.chk_auto_save.setToolTip(
-            "Automatically save work when using arrow keys to change images."
-        )
-        self.chk_auto_save.setChecked(True)
-        settings_layout.addWidget(self.chk_auto_save)
-
-        self.chk_save_npz = QCheckBox("Save .npz")
-        self.chk_save_npz.setChecked(True)
-        self.chk_save_npz.setToolTip(
-            "Save the final mask as a compressed NumPy NPZ file."
-        )
-        settings_layout.addWidget(self.chk_save_npz)
-
-        self.chk_save_txt = QCheckBox("Save .txt")
-        self.chk_save_txt.setChecked(True)
-        self.chk_save_txt.setToolTip(
-            "Save bounding box annotations in YOLO TXT format."
-        )
-        settings_layout.addWidget(self.chk_save_txt)
-
-        self.chk_yolo_use_alias = QCheckBox("Save YOLO with Class Aliases")
-        self.chk_yolo_use_alias.setToolTip(
-            "If checked, saves YOLO .txt files using class alias names instead of numeric IDs.\nThis is useful when a separate .yaml or .names file defines the classes."
-        )
-        self.chk_yolo_use_alias.setChecked(True)
-        settings_layout.addWidget(self.chk_yolo_use_alias)
-
-        self.chk_save_class_aliases = QCheckBox("Save Class Aliases (.json)")
-        self.chk_save_class_aliases.setToolTip(
-            "Save class aliases to a companion JSON file."
-        )
-        self.chk_save_class_aliases.setChecked(False)
-        settings_layout.addWidget(self.chk_save_class_aliases)
-
-        settings_group.setLayout(settings_layout)
-        main_layout.addWidget(settings_group)
-
-        sliders_group = QGroupBox("Adjustments")
-        sliders_layout = QVBoxLayout()
-
-        self.size_label = QLabel("Annotation Size: 1.0x")
-        self.size_slider = QSlider(Qt.Orientation.Horizontal)
-        self.size_slider.setRange(1, 50)
-        self.size_slider.setValue(10)
-        self.size_slider.setToolTip("Adjusts the size of points and lines (Ctrl +/-)")
-        sliders_layout.addWidget(self.size_label)
-        sliders_layout.addWidget(self.size_slider)
-
-        sliders_layout.addSpacing(10)
-
-        self.pan_label = QLabel("Pan Speed: 1.0x")
-        self.pan_slider = QSlider(Qt.Orientation.Horizontal)
-        self.pan_slider.setRange(1, 100)
-        self.pan_slider.setValue(10)
-        self.pan_slider.setToolTip(
-            "Adjusts the speed of WASD panning. Hold Shift for 5x boost."
-        )
-        sliders_layout.addWidget(self.pan_label)
-        sliders_layout.addWidget(self.pan_slider)
-
-        sliders_layout.addSpacing(10)
-
-        self.join_label = QLabel("Polygon Join Distance: 2px")
-        self.join_slider = QSlider(Qt.Orientation.Horizontal)
-        self.join_slider.setRange(1, 10)
-        self.join_slider.setValue(2)
-        self.join_slider.setToolTip("The pixel distance to 'snap' a polygon closed.")
-        sliders_layout.addWidget(self.join_label)
-        sliders_layout.addWidget(self.join_slider)
-
-        sliders_group.setLayout(sliders_layout)
-        main_layout.addWidget(sliders_group)
-
-        main_layout.addStretch()
-
-        self.notification_label = QLabel("")
-        font = self.notification_label.font()
-        font.setItalic(True)
-        self.notification_label.setFont(font)
-        self.notification_label.setStyleSheet("color: #ffa500;")
-        self.notification_label.setWordWrap(True)
-        main_layout.addWidget(self.notification_label)
-
-        self.device_label = QLabel("Device: Unknown")
-        main_layout.addWidget(self.device_label)
-
-        layout.addWidget(self.main_controls_widget)
-        self.setFixedWidth(250)
-
-
-class RightPanel(QWidget):
-    def __init__(self, parent=None):
-        super().__init__(parent)
-        self.v_layout = QVBoxLayout(self)
-
-        toggle_layout = QHBoxLayout()
-        toggle_layout.addStretch()
-        self.btn_toggle_visibility = QPushButton("Hide >")
-        self.btn_toggle_visibility.setToolTip("Hide this panel")
-        toggle_layout.addWidget(self.btn_toggle_visibility)
-        self.v_layout.addLayout(toggle_layout)
-
-        self.main_controls_widget = QWidget()
-        main_layout = QVBoxLayout(self.main_controls_widget)
-        main_layout.setContentsMargins(0, 0, 0, 0)
-
-        v_splitter = QSplitter(Qt.Orientation.Vertical)
-
-        file_explorer_widget = QWidget()
-        file_explorer_layout = QVBoxLayout(file_explorer_widget)
-        file_explorer_layout.setContentsMargins(0, 0, 0, 0)
-        self.btn_open_folder = QPushButton("Open Image Folder")
-        self.btn_open_folder.setToolTip("Open a directory of images")
-        self.file_tree = QTreeView()
-        file_explorer_layout.addWidget(self.btn_open_folder)
-        file_explorer_layout.addWidget(self.file_tree)
-        v_splitter.addWidget(file_explorer_widget)
-
-        segment_widget = QWidget()
-        segment_layout = QVBoxLayout(segment_widget)
-        segment_layout.setContentsMargins(0, 0, 0, 0)
-
-        class_filter_layout = QHBoxLayout()
-        class_filter_layout.addWidget(QLabel("Filter Class:"))
-        self.class_filter_combo = QComboBox()
-        self.class_filter_combo.setToolTip("Filter segments list by class")
-        class_filter_layout.addWidget(self.class_filter_combo)
-        segment_layout.addLayout(class_filter_layout)
-
-        self.segment_table = QTableWidget()
-        self.segment_table.setColumnCount(3)
-        self.segment_table.setHorizontalHeaderLabels(
-            ["Segment ID", "Class ID", "Alias"]
-        )
-        self.segment_table.horizontalHeader().setSectionResizeMode(
-            QHeaderView.ResizeMode.Stretch
-        )
-        self.segment_table.setSelectionBehavior(
-            QAbstractItemView.SelectionBehavior.SelectRows
-        )
-        self.segment_table.setSortingEnabled(True)
-        self.segment_table.setEditTriggers(QAbstractItemView.EditTrigger.NoEditTriggers)
-        segment_layout.addWidget(self.segment_table)
-
-        segment_action_layout = QHBoxLayout()
-        self.btn_merge_selection = QPushButton("Merge to Class")
-        self.btn_merge_selection.setToolTip(
-            "Merge selected segments into a single class (M)"
-        )
-        self.btn_delete_selection = QPushButton("Delete")
-        self.btn_delete_selection.setToolTip(
-            "Delete selected segments (Delete/Backspace)"
-        )
-        segment_action_layout.addWidget(self.btn_merge_selection)
-        segment_action_layout.addWidget(self.btn_delete_selection)
-        segment_layout.addLayout(segment_action_layout)
-        v_splitter.addWidget(segment_widget)
-
-        class_widget = QWidget()
-        class_layout = QVBoxLayout(class_widget)
-        class_layout.setContentsMargins(0, 0, 0, 0)
-        class_layout.addWidget(QLabel("Class Order:"))
-        self.class_table = ReorderableClassTable()
-        self.class_table.setToolTip(
-            "Double-click to set class aliases and drag to reorder channels for saving."
-        )
-        self.class_table.setColumnCount(2)
-        self.class_table.setHorizontalHeaderLabels(["Alias", "Class ID"])
-        self.class_table.horizontalHeader().setSectionResizeMode(
-            0, QHeaderView.ResizeMode.Stretch
-        )
-        self.class_table.horizontalHeader().setSectionResizeMode(
-            1, QHeaderView.ResizeMode.ResizeToContents
-        )
-        self.class_table.setEditTriggers(QAbstractItemView.EditTrigger.DoubleClicked)
-        class_layout.addWidget(self.class_table)
-        self.btn_reassign_classes = QPushButton("Reassign Class IDs")
-        self.btn_reassign_classes.setToolTip(
-            "Re-index class channels based on the current order in this table"
-        )
-        class_layout.addWidget(self.btn_reassign_classes)
-        v_splitter.addWidget(class_widget)
-
-        main_layout.addWidget(v_splitter)
-
-        self.status_label = QLabel("")
-        self.status_label.setAlignment(Qt.AlignmentFlag.AlignCenter)
-        main_layout.addWidget(self.status_label)
-
-        self.v_layout.addWidget(self.main_controls_widget)
-        self.setFixedWidth(350)

lazylabel/sam_model.py

@@ -1,70 +0,0 @@
-import os
-import cv2
-import numpy as np
-import torch
-import requests
-from tqdm import tqdm
-from segment_anything import sam_model_registry, SamPredictor
-
-
-def download_model(url, download_path):
-    """Downloads file with a progress bar."""
-    print(
-        f"SAM model not found. Downloading from Meta's GitHub repository to: {download_path}"
-    )
-    response = requests.get(url, stream=True)
-    response.raise_for_status()
-    total_size_in_bytes = int(response.headers.get("content-length", 0))
-    block_size = 1024  # 1 Kibibyte
-
-    progress_bar = tqdm(total=total_size_in_bytes, unit="iB", unit_scale=True)
-    with open(download_path, "wb") as file:
-        for data in response.iter_content(block_size):
-            progress_bar.update(len(data))
-            file.write(data)
-    progress_bar.close()
-
-    if total_size_in_bytes != 0 and progress_bar.n != total_size_in_bytes:
-        print("ERROR, something went wrong during download")
-
-
-class SamModel:
-    def __init__(self, model_type, model_filename="sam_vit_h_4b8939.pth"):
-        self.device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
-
-        # Define model URL and local cache path
-        model_url = f"https://dl.fbaipublicfiles.com/segment_anything/{model_filename}"
-        cache_dir = os.path.join(os.path.expanduser("~"), ".cache", "lazylabel")
-        os.makedirs(cache_dir, exist_ok=True)
-        model_path = os.path.join(cache_dir, model_filename)
-
-        # Download the model if it doesn't exist
-        if not os.path.exists(model_path):
-            download_model(model_url, model_path)
-
-        print(f"Loading SAM model from {model_path}...")
-        self.model = sam_model_registry[model_type](checkpoint=model_path).to(
-            self.device
-        )
-        self.predictor = SamPredictor(self.model)
-        self.image = None
-        print("SAM model loaded successfully.")
-
-    def set_image(self, image_path):
-        self.image = cv2.imread(image_path)
-        self.image = cv2.cvtColor(self.image, cv2.COLOR_BGR2RGB)
-        self.predictor.set_image(self.image)
-
-    def predict(self, positive_points, negative_points):
-        if not positive_points:
-            return None
-
-        points = np.array(positive_points + negative_points)
-        labels = np.array([1] * len(positive_points) + [0] * len(negative_points))
-
-        masks, _, _ = self.predictor.predict(
-            point_coords=points,
-            point_labels=labels,
-            multimask_output=False,
-        )
-        return masks[0]

lazylabel_gui-1.0.9.dist-info/RECORD

@@ -1,17 +0,0 @@
-lazylabel/controls.py,sha256=y983bXRaeWwrCOdpxAO2zjXGv1SM8zUSrJjUL6sbss4,10832
-lazylabel/custom_file_system_model.py,sha256=We0Q1f3c_mg0s5Du44NspjzuIQf8Cvgy7t8jQ2c53z0,4652
-lazylabel/editable_vertex.py,sha256=itGcZG5MyuctGfxjINu8IJBYFFsCGuE_YtsrfHICjiw,1115
-lazylabel/hoverable_pixelmap_item.py,sha256=kJFOp7WXiyHpNf7l73TZjiob85jgP30b5MZvu_z5L3c,728
-lazylabel/hoverable_polygon_item.py,sha256=-0l8C8PfsXtJGqvZZ2qtizxHmFwO8RCwz5UfjKpDvzY,775
-lazylabel/main.py,sha256=Iq1_2C2j1W3oghr1pYs2SROQIPmNf5HBOvKaLcC-uD4,51663
-lazylabel/numeric_table_widget_item.py,sha256=dQUlIFu9syCxTGAHVIlmbgkI7aJ3f3wmDPBz1AGK9Bg,283
-lazylabel/photo_viewer.py,sha256=PNgm0gU2gnIqvRkrGlQugdobGsKwAi3m3X6ZF487lCo,2055
-lazylabel/reorderable_class_table.py,sha256=4c-iuSkPcmk5Aey5n2zz49O85x9TQPujKG-JLxtuBCo,2406
-lazylabel/sam_model.py,sha256=9NB51Xq1P5dIxZMBdttwwRlszlJR3U5HRs83QsPLxNE,2595
-lazylabel/utils.py,sha256=sYSCoXL27OaLgOZaUkCAhgmKZ7YfhR3Cc5F8nDIa3Ig,414
-lazylabel_gui-1.0.9.dist-info/licenses/LICENSE,sha256=kSDEIgrWAPd1u2UFGGpC9X71dhzrlzBFs8hbDlENnGE,1092
-lazylabel_gui-1.0.9.dist-info/METADATA,sha256=9_fhalqkG4brIXovWfgoAfIV-AyRyrr5v0CSgr3yEzA,6317
-lazylabel_gui-1.0.9.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-lazylabel_gui-1.0.9.dist-info/entry_points.txt,sha256=Hd0WwEG9OPTa_ziYjiD0aRh7R6Fupt-wdQ3sspdc1mM,54
-lazylabel_gui-1.0.9.dist-info/top_level.txt,sha256=YN4uIyrpDBq1wiJaBuZLDipIzyZY0jqJOmmXiPIOUkU,10
-lazylabel_gui-1.0.9.dist-info/RECORD,,