lazylabel-gui 1.0.8__tar.gz → 1.0.9__tar.gz

{lazylabel_gui-1.0.8 → lazylabel_gui-1.0.9}/PKG-INFO

@@ -1,7 +1,7 @@
 Metadata-Version: 2.4
 Name: lazylabel-gui
-Version: 1.0.8
-Summary: An image segmentation GUI for generating mask tensors.
+Version: 1.0.9
+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
         

{lazylabel_gui-1.0.8 → lazylabel_gui-1.0.9}/pyproject.toml

@@ -4,11 +4,11 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "lazylabel-gui"
-version = "1.0.8"
+version = "1.0.9"
 authors = [
   { name="Deniz N. Cakan", email="deniz.n.cakan@gmail.com" },
 ]
-description = "An image segmentation GUI for generating mask tensors."
+description = "An image segmentation GUI for generating ML ready mask tensors and annotations."
 readme = "README.md"
 license = { file="LICENSE" }
 requires-python = ">=3.10"

{lazylabel_gui-1.0.8 → lazylabel_gui-1.0.9}/src/lazylabel/controls.py

@@ -92,6 +92,13 @@ class ControlPanel(QWidget):
         )
         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."
@@ -173,7 +180,6 @@ class RightPanel(QWidget):
 
         v_splitter = QSplitter(Qt.Orientation.Vertical)
 
-        # --- File Explorer Widget ---
         file_explorer_widget = QWidget()
         file_explorer_layout = QVBoxLayout(file_explorer_widget)
         file_explorer_layout.setContentsMargins(0, 0, 0, 0)
@@ -184,7 +190,6 @@ class RightPanel(QWidget):
         file_explorer_layout.addWidget(self.file_tree)
         v_splitter.addWidget(file_explorer_widget)
 
-        # --- Segment List Widget ---
         segment_widget = QWidget()
         segment_layout = QVBoxLayout(segment_widget)
         segment_layout.setContentsMargins(0, 0, 0, 0)
@@ -225,7 +230,6 @@ class RightPanel(QWidget):
         segment_layout.addLayout(segment_action_layout)
         v_splitter.addWidget(segment_widget)
 
-        # --- Class Table Widget ---
         class_widget = QWidget()
         class_layout = QVBoxLayout(class_widget)
         class_layout.setContentsMargins(0, 0, 0, 0)

lazylabel_gui-1.0.9/src/lazylabel/custom_file_system_model.py

@@ -0,0 +1,126 @@
+from pathlib import Path
+from PyQt6.QtCore import Qt, QModelIndex, QDir
+from PyQt6.QtGui import QFileSystemModel, QBrush, QColor
+
+
+class CustomFileSystemModel(QFileSystemModel):
+    def __init__(self, parent=None):
+        super().__init__(parent)
+        self.setFilter(QDir.Filter.NoDotAndDotDot | QDir.Filter.Files)
+        self.setNameFilterDisables(False)
+        self.setNameFilters(["*.png", "*.jpg", "*.jpeg", "*.tiff", "*.tif"])
+        self.highlighted_path = None
+
+        self.npz_files = set()
+        self.txt_files = set()
+
+    def setRootPath(self, path: str) -> QModelIndex:
+        self._scan_directory(path)
+        return super().setRootPath(path)
+
+    def _scan_directory(self, path: str):
+        """Scans the directory once and caches the basenames of .npz and .txt files."""
+        self.npz_files.clear()
+        self.txt_files.clear()
+        if not path:
+            return
+
+        directory = Path(path)
+        if not directory.is_dir():
+            return
+
+        try:
+            for file_path in directory.iterdir():
+                if file_path.suffix == ".npz":
+                    self.npz_files.add(file_path.stem)
+                elif file_path.suffix == ".txt":
+                    self.txt_files.add(file_path.stem)
+        except OSError:
+            pass
+
+    def update_cache_for_path(self, saved_file_path: str):
+        """Incrementally updates the cache and the view for a newly saved file."""
+        if not saved_file_path:
+            return
+
+        p = Path(saved_file_path)
+        base_name = p.stem
+
+        if p.suffix == ".npz":
+            self.npz_files.add(base_name)
+        elif p.suffix == ".txt":
+            self.txt_files.add(base_name)
+        else:
+            return
+
+        # Find the model index for the corresponding image file to refresh its row
+        # This assumes the image file is in the same directory (the root path)
+        root_path = Path(self.rootPath())
+        for image_ext in self.nameFilters():  # e.g., '*.png', '*.jpg'
+            # Construct full path to the potential image file
+            image_file = root_path / (base_name + image_ext.replace("*", ""))
+            index = self.index(str(image_file))
+
+            if index.isValid() and index.row() != -1:
+                # Found the corresponding image file, emit signal to refresh its checkmarks
+                index_col1 = self.index(index.row(), 1, index.parent())
+                index_col2 = self.index(index.row(), 2, index.parent())
+                self.dataChanged.emit(
+                    index_col1, index_col2, [Qt.ItemDataRole.CheckStateRole]
+                )
+                break
+
+    def set_highlighted_path(self, path):
+        self.highlighted_path = str(Path(path)) if path else None
+        self.layoutChanged.emit()
+
+    def columnCount(self, parent: QModelIndex = QModelIndex()) -> int:
+        return 3
+
+    def headerData(
+        self,
+        section: int,
+        orientation: Qt.Orientation,
+        role: int = Qt.ItemDataRole.DisplayRole,
+    ):
+        if (
+            orientation == Qt.Orientation.Horizontal
+            and role == Qt.ItemDataRole.DisplayRole
+        ):
+            if section == 0:
+                return "File Name"
+            if section == 1:
+                return ".npz"
+            if section == 2:
+                return ".txt"
+        return super().headerData(section, orientation, role)
+
+    def data(self, index: QModelIndex, role: int = Qt.ItemDataRole.DisplayRole):
+        if not index.isValid():
+            return None
+
+        if role == Qt.ItemDataRole.BackgroundRole:
+            filePath = self.filePath(index)
+            if self.highlighted_path:
+                p_file = Path(filePath)
+                p_highlight = Path(self.highlighted_path)
+                if p_file.with_suffix("") == p_highlight.with_suffix(""):
+                    return QBrush(QColor(40, 80, 40))
+
+        if index.column() > 0 and role == Qt.ItemDataRole.CheckStateRole:
+            fileName = self.fileName(index.siblingAtColumn(0))
+            base_name = Path(fileName).stem
+
+            if index.column() == 1:
+                exists = base_name in self.npz_files
+            elif index.column() == 2:
+                exists = base_name in self.txt_files
+            else:
+                return None
+
+            return Qt.CheckState.Checked if exists else Qt.CheckState.Unchecked
+
+        if index.column() > 0 and role == Qt.ItemDataRole.DisplayRole:
+            return ""
+
+        return super().data(index, role)

{lazylabel_gui-1.0.8 → lazylabel_gui-1.0.9}/src/lazylabel/main.py

@@ -1016,13 +1016,27 @@ class MainWindow(QMainWindow):
                         np.savez_compressed(
                             npz_path, mask=final_mask_tensor.astype(np.uint8)
                         )
+                        self.file_model.update_cache_for_path(npz_path)
+
                         self.file_model.set_highlighted_path(npz_path)
                         QTimer.singleShot(
                             1500, lambda: self.file_model.set_highlighted_path(None)
                         )
                         saved_something = True
                     if save_txt:
-                        self.generate_yolo_annotations(final_mask_tensor)
+                        if self.control_panel.chk_yolo_use_alias.isChecked():
+                            class_labels = [
+                                class_table.item(row, 0).text()
+                                for row in range(class_table.rowCount())
+                            ]
+                        else:
+                            class_labels = list(range(num_final_classes))
+
+                        txt_path = self.generate_yolo_annotations(
+                            final_mask_tensor, class_labels
+                        )
+                        if txt_path:
+                            self.file_model.update_cache_for_path(txt_path)
                         saved_something = True
 
         if save_aliases:
@@ -1039,7 +1053,7 @@ class MainWindow(QMainWindow):
 
         QTimer.singleShot(3000, lambda: self.right_panel.status_label.clear())
 
-    def generate_yolo_annotations(self, mask_tensor):
+    def generate_yolo_annotations(self, mask_tensor, class_labels):
         output_path = os.path.splitext(self.current_image_path)[0] + ".txt"
         h, w, num_channels = mask_tensor.shape
 
@@ -1055,20 +1069,26 @@ class MainWindow(QMainWindow):
             contours, _ = cv2.findContours(
                 single_channel_image, cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE
             )
-            class_id = channel
+
+            class_label = class_labels[channel]
             for contour in contours:
                 x, y, width, height = cv2.boundingRect(contour)
                 center_x = (x + width / 2) / w
                 center_y = (y + height / 2) / h
                 normalized_width = width / w
                 normalized_height = height / h
-                yolo_entry = f"{class_id} {center_x} {center_y} {normalized_width} {normalized_height}"
+                yolo_entry = f"{class_label} {center_x} {center_y} {normalized_width} {normalized_height}"
                 yolo_annotations.append(yolo_entry)
 
+        if not yolo_annotations:
+            return None
+
         with open(output_path, "w") as file:
             for annotation in yolo_annotations:
                 file.write(annotation + "\n")
 
+        return output_path
+
     def save_current_segment(self):
         if (
             self.mode != "sam_points"

{lazylabel_gui-1.0.8 → lazylabel_gui-1.0.9}/src/lazylabel_gui.egg-info/PKG-INFO

@@ -1,7 +1,7 @@
 Metadata-Version: 2.4
 Name: lazylabel-gui
-Version: 1.0.8
-Summary: An image segmentation GUI for generating mask tensors.
+Version: 1.0.9
+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
         

lazylabel_gui-1.0.8/src/lazylabel/custom_file_system_model.py

@@ -1,72 +0,0 @@
-import os
-from PyQt6.QtCore import Qt, QModelIndex, QDir
-from PyQt6.QtGui import QFileSystemModel, QBrush, QColor
-
-
-class CustomFileSystemModel(QFileSystemModel):
-    def __init__(self, parent=None):
-        super().__init__(parent)
-        self.setFilter(QDir.Filter.NoDotAndDotDot | QDir.Filter.Files)
-        self.setNameFilterDisables(False)
-        self.setNameFilters(["*.png", "*.jpg", "*.jpeg", "*.tiff", "*.tif"])
-        self.highlighted_path = None
-
-    def set_highlighted_path(self, path):
-        self.highlighted_path = os.path.normpath(path) if path else None
-        self.layoutChanged.emit()
-
-    def columnCount(self, parent: QModelIndex = QModelIndex()) -> int:
-        return 3
-
-    def headerData(
-        self,
-        section: int,
-        orientation: Qt.Orientation,
-        role: int = Qt.ItemDataRole.DisplayRole,
-    ):
-        if (
-            orientation == Qt.Orientation.Horizontal
-            and role == Qt.ItemDataRole.DisplayRole
-        ):
-            if section == 0:
-                return "File Name"
-            if section == 1:
-                return ".npz"
-            if section == 2:
-                return ".txt"
-        return super().headerData(section, orientation, role)
-
-    def data(self, index: QModelIndex, role: int = Qt.ItemDataRole.DisplayRole):
-        if not index.isValid():
-            return None
-
-        if role == Qt.ItemDataRole.BackgroundRole:
-            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))
-
-        if index.column() > 0 and role == Qt.ItemDataRole.CheckStateRole:
-            filePath = self.filePath(index.siblingAtColumn(0))
-            base_path = os.path.splitext(filePath)[0]
-
-            if index.column() == 1:
-                check_path = base_path + ".npz"
-            elif index.column() == 2:
-                check_path = base_path + ".txt"
-            else:
-                return None
-
-            return (
-                Qt.CheckState.Checked
-                if os.path.exists(check_path)
-                else Qt.CheckState.Unchecked
-            )
-
-        if index.column() > 0 and role == Qt.ItemDataRole.DisplayRole:
-            return ""
-
-        return super().data(index, role)