lazylabel-gui 1.0.6__tar.gz → 1.0.7__tar.gz

{lazylabel_gui-1.0.6 → lazylabel_gui-1.0.7}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: lazylabel-gui
-Version: 1.0.6
+Version: 1.0.7
 Summary: An image segmentation GUI for generating mask tensors.
 Author-email: "Deniz N. Cakan" <deniz.n.cakan@gmail.com>
 License: MIT License
@@ -48,7 +48,7 @@ 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 format for easy machine learning integration.
+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.
 
 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).
 

{lazylabel_gui-1.0.6 → lazylabel_gui-1.0.7}/README.md

@@ -1,5 +1,5 @@
 # <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 format for easy machine learning integration.
+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.
 
 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).
 

{lazylabel_gui-1.0.6 → lazylabel_gui-1.0.7}/pyproject.toml

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

{lazylabel_gui-1.0.6 → lazylabel_gui-1.0.7}/src/lazylabel/controls.py

@@ -10,6 +10,7 @@ from PyQt6.QtWidgets import (
     QHBoxLayout,
     QComboBox,
     QHeaderView,
+    QCheckBox,
 )
 from PyQt6.QtCore import Qt
 from .reorderable_class_table import ReorderableClassTable
@@ -26,20 +27,56 @@ class ControlPanel(QWidget):
         font.setBold(True)
         self.mode_label.setFont(font)
         layout.addWidget(self.mode_label)
+
+        # Mode Buttons
         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)")
         layout.addWidget(self.btn_sam_mode)
         layout.addWidget(self.btn_polygon_mode)
         layout.addWidget(self.btn_selection_mode)
+
         layout.addSpacing(20)
         line1 = QFrame()
         line1.setFrameShape(QFrame.Shape.HLine)
         layout.addWidget(line1)
         layout.addSpacing(10)
+
+        # Action Buttons
+        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)")
+        layout.addWidget(self.btn_fit_view)
         layout.addWidget(self.btn_clear_points)
+
+        layout.addSpacing(10)
+
+        # Settings
+        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)
+        layout.addWidget(self.chk_auto_save)
+
         layout.addStretch()
+
+        # Notification Label
+        self.notification_label = QLabel("")
+        font = self.notification_label.font()
+        font.setItalic(True)
+        self.notification_label.setFont(font)
+        self.notification_label.setStyleSheet(
+            "color: #ffa500;"
+        )  # Orange color for visibility
+        self.notification_label.setWordWrap(True)
+        layout.addWidget(self.notification_label)
+
+        # Device Label
         self.device_label = QLabel("Device: Unknown")
         layout.addWidget(self.device_label)
         self.setFixedWidth(250)
@@ -53,6 +90,7 @@ class RightPanel(QWidget):
         # File Explorer
         file_explorer_layout = QVBoxLayout()
         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)
@@ -68,12 +106,13 @@ class RightPanel(QWidget):
         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(["Index", "Class", "Type"])
+        self.segment_table.setHorizontalHeaderLabels(["Index", "Class ID", "Type"])
         self.segment_table.horizontalHeader().setSectionResizeMode(
             QHeaderView.ResizeMode.Stretch
         )
@@ -85,7 +124,13 @@ class RightPanel(QWidget):
 
         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)
@@ -95,13 +140,23 @@ class RightPanel(QWidget):
         class_layout = QVBoxLayout()
         class_layout.addWidget(QLabel("Class Order:"))
         self.class_table = ReorderableClassTable()
-        self.class_table.setColumnCount(1)
-        self.class_table.setHorizontalHeaderLabels(["Class ID"])
+        self.class_table.setToolTip(
+            "Set class aliases and drag to reorder channels for saving."
+        )
+        self.class_table.setColumnCount(2)
+        self.class_table.setHorizontalHeaderLabels(["Alias", "Channel Index"])
         self.class_table.horizontalHeader().setSectionResizeMode(
-            QHeaderView.ResizeMode.Stretch
+            0, QHeaderView.ResizeMode.Stretch
         )
+        self.class_table.horizontalHeader().setSectionResizeMode(
+            1, QHeaderView.ResizeMode.ResizeToContents
+        )
+
         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)
         layout.addLayout(class_layout, 1)
 

{lazylabel_gui-1.0.6 → lazylabel_gui-1.0.7}/src/lazylabel/custom_file_system_model.py

@@ -12,7 +12,9 @@ class CustomFileSystemModel(QFileSystemModel):
         self.highlighted_path = None
 
     def set_highlighted_path(self, path):
-        self.highlighted_path = path
+        self.highlighted_path = os.path.normpath(path) if path else None
+        # Trigger repaint of the entire view
+        self.layoutChanged.emit()
 
     def columnCount(self, parent: QModelIndex = QModelIndex()) -> int:
         return 2
@@ -39,9 +41,13 @@ class CustomFileSystemModel(QFileSystemModel):
 
         # Handle the temporary highlight for saving
         if role == Qt.ItemDataRole.BackgroundRole:
-            filePath = self.filePath(index)
-            if filePath == self.highlighted_path:
-                return QBrush(QColor("yellow"))
+            filePath = os.path.normpath(self.filePath(index))
+            if (
+                self.highlighted_path
+                and os.path.splitext(filePath)[0]
+                == os.path.splitext(self.highlighted_path)[0]
+            ):
+                return QBrush(QColor(40, 80, 40))  # Dark green highlight
 
         if index.column() == 1:
             if role == Qt.ItemDataRole.CheckStateRole:

{lazylabel_gui-1.0.6 → lazylabel_gui-1.0.7}/src/lazylabel/editable_vertex.py

@@ -1,6 +1,6 @@
 from PyQt6.QtWidgets import QGraphicsEllipseItem, QGraphicsItem
 from PyQt6.QtCore import Qt
-from PyQt6.QtGui import QBrush, QPen
+from PyQt6.QtGui import QBrush, QPen, QColor
 
 
 class EditableVertexItem(QGraphicsEllipseItem):
@@ -11,8 +11,12 @@ class EditableVertexItem(QGraphicsEllipseItem):
         self.vertex_index = vertex_index
 
         self.setZValue(200)
-        self.setBrush(QBrush(Qt.GlobalColor.cyan))
-        self.setPen(QPen(Qt.GlobalColor.white, 1))
+
+        color = QColor(Qt.GlobalColor.cyan)
+        color.setAlpha(180)
+        self.setBrush(QBrush(color))
+
+        self.setPen(QPen(Qt.GlobalColor.transparent))
         self.setFlag(QGraphicsItem.GraphicsItemFlag.ItemIsMovable)
         self.setFlag(QGraphicsItem.GraphicsItemFlag.ItemSendsGeometryChanges)
 

lazylabel_gui-1.0.7/src/lazylabel/hoverable_pixelmap_item.py

@@ -0,0 +1,22 @@
+from PyQt6.QtWidgets import QGraphicsPixmapItem
+
+
+class HoverablePixmapItem(QGraphicsPixmapItem):
+    def __init__(self, parent=None):
+        super().__init__(parent)
+        self.setAcceptHoverEvents(True)
+        self.default_pixmap = None
+        self.hover_pixmap = None
+
+    def set_pixmaps(self, default_pixmap, hover_pixmap):
+        self.default_pixmap = default_pixmap
+        self.hover_pixmap = hover_pixmap
+        self.setPixmap(self.default_pixmap)
+
+    def hoverEnterEvent(self, event):
+        self.setPixmap(self.hover_pixmap)
+        super().hoverEnterEvent(event)
+
+    def hoverLeaveEvent(self, event):
+        self.setPixmap(self.default_pixmap)
+        super().hoverLeaveEvent(event)