lazylabel-gui 1.0.2__tar.gz → 1.0.4__tar.gz

{lazylabel_gui-1.0.2 → lazylabel_gui-1.0.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: lazylabel-gui
-Version: 1.0.2
+Version: 1.0.4
 Summary: An image segmentation GUI for generating mask tensors.
 Author-email: "Deniz N. Cakan" <deniz.n.cakan@gmail.com>
 License: MIT License
@@ -70,11 +70,11 @@ Inspired by [LabelMe](https://github.com/wkentaro/labelme?tab=readme-ov-file#ins
 ## 🚀 Getting Started
 
 ### Prerequisites
-Ensure you have **Python 3.10** or newer.
+**Python 3.10**
 
 ### Installation
 
-#### For Users (via PyPI)
+#### For Users [via PyPI](https://pypi.org/project/lazylabel-gui/)
 1.  Install LazyLabel directly:
     ```bash
     pip install lazylabel-gui
@@ -87,7 +87,7 @@ Ensure you have **Python 3.10** or newer.
 #### For Developers (from Source)
 1.  Clone the repository:
     ```bash
-    git clone [https://github.com/dnzckn/LazyLabel.git](https://github.com/dnzckn/LazyLabel.git)
+    git clone https://github.com/dnzckn/LazyLabel.git
     cd LazyLabel
     ```
 2.  Install in editable mode, which links the installed package to your source directory:

{lazylabel_gui-1.0.2 → lazylabel_gui-1.0.4}/README.md

@@ -21,11 +21,11 @@ Inspired by [LabelMe](https://github.com/wkentaro/labelme?tab=readme-ov-file#ins
 ## 🚀 Getting Started
 
 ### Prerequisites
-Ensure you have **Python 3.10** or newer.
+**Python 3.10**
 
 ### Installation
 
-#### For Users (via PyPI)
+#### For Users [via PyPI](https://pypi.org/project/lazylabel-gui/)
 1.  Install LazyLabel directly:
     ```bash
     pip install lazylabel-gui
@@ -38,7 +38,7 @@ Ensure you have **Python 3.10** or newer.
 #### For Developers (from Source)
 1.  Clone the repository:
     ```bash
-    git clone [https://github.com/dnzckn/LazyLabel.git](https://github.com/dnzckn/LazyLabel.git)
+    git clone https://github.com/dnzckn/LazyLabel.git
     cd LazyLabel
     ```
 2.  Install in editable mode, which links the installed package to your source directory:
@@ -95,4 +95,4 @@ Each channel is a binary mask for a class, combining all assigned segments into
 ---
 
 ## ☕ Support LazyLabel
-[If you found LazyLabel helpful, consider supporting the project!](https://buymeacoffee.com/dnzckn)
+[If you found LazyLabel helpful, consider supporting the project!](https://buymeacoffee.com/dnzckn)

{lazylabel_gui-1.0.2 → lazylabel_gui-1.0.4}/pyproject.toml

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

{lazylabel_gui-1.0.2 → lazylabel_gui-1.0.4}/src/lazylabel/main.py

@@ -19,7 +19,6 @@ from PyQt6.QtWidgets import (
 from PyQt6.QtGui import QPixmap, QColor, QPen, QBrush, QPolygonF, QIcon
 from PyQt6.QtCore import Qt, QPointF, QTimer
 
-# Relative imports for package structure
 from .photo_viewer import PhotoViewer
 from .sam_model import SamModel
 from .utils import mask_to_pixmap
@@ -43,7 +42,6 @@ class MainWindow(QMainWindow):
 
         self.setGeometry(50, 50, 1600, 900)
 
-        # The SamModel instance is now passed in
         self.sam_model = sam_model
         self.mode = "sam_points"
         self.previous_mode = "sam_points"
@@ -117,11 +115,26 @@ class MainWindow(QMainWindow):
         )
         self.control_panel.btn_clear_points.clicked.connect(self.clear_all_points)
 
+    def _get_color_for_class(self, class_id, saturation, value):
+        if class_id is None:
+            return QColor.fromHsv(0, 0, 128)
+
+        hue = int((class_id * 222.4922359) % 360)
+        color = QColor.fromHsv(hue, saturation, value)
+
+        if not color.isValid():
+            return QColor(Qt.GlobalColor.white)
+        return color
+
     def set_mode(self, mode_name, is_toggle=False):
+        if self.mode == "selection" and mode_name not in ["selection", "edit"]:
+            self.right_panel.segment_table.clearSelection()
         if self.mode == "edit" and mode_name != "edit":
             self.display_all_segments()
-        if not is_toggle and self.mode not in ["pan", "selection", "edit"]:
+
+        if not is_toggle and self.mode not in ["selection", "edit"]:
             self.previous_mode = self.mode
+
         self.mode = mode_name
         self.control_panel.mode_label.setText(
             f"Mode: {mode_name.replace('_', ' ').title()}"
@@ -143,7 +156,7 @@ class MainWindow(QMainWindow):
         if self.mode == new_mode:
             self.set_mode(self.previous_mode, is_toggle=True)
         else:
-            if self.mode not in ["pan", "selection", "edit"]:
+            if self.mode not in ["selection", "edit"]:
                 self.previous_mode = self.mode
             self.set_mode(new_mode, is_toggle=True)
 
@@ -179,7 +192,9 @@ class MainWindow(QMainWindow):
         self.current_file_index = index
         path = self.file_model.filePath(index)
 
-        if os.path.isfile(path) and path.lower().endswith((".png", ".jpg", ".jpeg")):
+        if os.path.isfile(path) and path.lower().endswith(
+            (".png", ".jpg", ".jpeg", ".tiff")
+        ):
             self.current_image_path = path
             pixmap = QPixmap(self.current_image_path)
             if not pixmap.isNull():
@@ -238,12 +253,13 @@ class MainWindow(QMainWindow):
             self.toggle_pan_mode()
         elif key == Qt.Key.Key_R:
             self.toggle_edit_mode()
-        elif key == Qt.Key.Key_C:
+        elif key == Qt.Key.Key_C or key == Qt.Key.Key_Escape:
             self.clear_all_points()
         elif key == Qt.Key.Key_V or key == Qt.Key.Key_Backspace:
             self.delete_selected_segments()
         elif key == Qt.Key.Key_M:
             self.assign_selected_to_class()
+            self.right_panel.segment_table.clearSelection()
         elif key == Qt.Key.Key_Z and mods == Qt.KeyboardModifier.ControlModifier:
             self.undo_last_action()
         elif key == Qt.Key.Key_A and mods == Qt.KeyboardModifier.ControlModifier:
@@ -387,10 +403,23 @@ class MainWindow(QMainWindow):
         selected_indices = self.get_selected_segment_indices()
         if not selected_indices:
             return
-        target_class_id = self.segments[selected_indices[0]]["class_id"]
+
+        existing_class_ids = [
+            self.segments[i]["class_id"]
+            for i in selected_indices
+            if self.segments[i].get("class_id") is not None
+        ]
+
+        if existing_class_ids:
+            target_class_id = min(existing_class_ids)
+        else:
+            target_class_id = self.segments[selected_indices[0]].get("class_id")
+
         for i in selected_indices:
             self.segments[i]["class_id"] = target_class_id
+
         self.update_all_lists()
+        self.right_panel.segment_table.clearSelection()
         self.viewer.setFocus()
 
     def rasterize_polygon(self, vertices):
@@ -412,26 +441,10 @@ class MainWindow(QMainWindow):
         self.segment_items.clear()
         selected_indices = self.get_selected_segment_indices()
 
-        unique_class_ids = sorted(
-            list(
-                {
-                    seg.get("class_id")
-                    for seg in self.segments
-                    if seg.get("class_id") is not None
-                }
-            )
-        )
-        num_classes = len(unique_class_ids) if unique_class_ids else 1
-        class_id_to_hue_index = {
-            class_id: i for i, class_id in enumerate(unique_class_ids)
-        }
-
         for i, seg_dict in enumerate(self.segments):
             self.segment_items[i] = []
-            class_id = seg_dict.get("class_id", 0)
-            hue_index = class_id_to_hue_index.get(class_id, 0)
-            hue = int((hue_index * 360 / num_classes)) % 360
-            base_color = QColor.fromHsv(hue, 220, 220)
+            class_id = seg_dict.get("class_id")
+            base_color = self._get_color_for_class(class_id, saturation=220, value=220)
 
             if seg_dict["type"] == "Polygon":
                 poly_item = HoverablePolygonItem(QPolygonF(seg_dict["vertices"]))
@@ -522,27 +535,14 @@ class MainWindow(QMainWindow):
 
         table.setRowCount(len(display_segments))
 
-        unique_class_ids = sorted(
-            list(
-                {
-                    s.get("class_id")
-                    for s in self.segments
-                    if s.get("class_id") is not None
-                }
-            )
-        )
-        num_classes = len(unique_class_ids) if unique_class_ids else 1
-        class_id_to_hue_index = {cid: i for i, cid in enumerate(unique_class_ids)}
-
         for row, (original_index, seg) in enumerate(display_segments):
-            class_id = seg.get("class_id", 0)
-            hue_index = class_id_to_hue_index.get(class_id, 0)
-            hue = int((hue_index * 360 / num_classes)) % 360
-            color = QColor.fromHsv(hue, 150, 100)
+            class_id = seg.get("class_id")
+            color = self._get_color_for_class(class_id, saturation=180, value=200)
 
+            class_id_str = str(class_id) if class_id is not None else "N/A"
             index_item = NumericTableWidgetItem(str(original_index + 1))
-            class_item = NumericTableWidgetItem(str(class_id))
-            type_item = QTableWidgetItem(seg["type"])
+            class_item = NumericTableWidgetItem(class_id_str)
+            type_item = QTableWidgetItem(seg.get("type", "N/A"))
 
             index_item.setFlags(index_item.flags() & ~Qt.ItemFlag.ItemIsEditable)
             type_item.setFlags(type_item.flags() & ~Qt.ItemFlag.ItemIsEditable)
@@ -551,12 +551,15 @@ class MainWindow(QMainWindow):
             table.setItem(row, 0, index_item)
             table.setItem(row, 1, class_item)
             table.setItem(row, 2, type_item)
+
             for col in range(3):
-                table.item(row, col).setBackground(QBrush(color))
+                if table.item(row, col):
+                    table.item(row, col).setBackground(QBrush(color))
 
         table.setSortingEnabled(False)
         for row in range(table.rowCount()):
-            if table.item(row, 0).data(Qt.ItemDataRole.UserRole) in selected_indices:
+            item = table.item(row, 0)
+            if item and item.data(Qt.ItemDataRole.UserRole) in selected_indices:
                 table.selectRow(row)
         table.setSortingEnabled(True)
 
@@ -566,6 +569,8 @@ class MainWindow(QMainWindow):
     def update_class_list(self):
         class_table = self.right_panel.class_table
         class_table.blockSignals(True)
+        class_table.clearContents()
+
         unique_class_ids = sorted(
             list(
                 {
@@ -576,18 +581,16 @@ class MainWindow(QMainWindow):
             )
         )
         class_table.setRowCount(len(unique_class_ids))
-        num_classes = len(unique_class_ids) if unique_class_ids else 1
-        class_id_to_hue_index = {
-            class_id: i for i, class_id in enumerate(unique_class_ids)
-        }
+
         for row, cid in enumerate(unique_class_ids):
             item = QTableWidgetItem(str(cid))
             item.setFlags(item.flags() & ~Qt.ItemFlag.ItemIsEditable)
-            hue_index = class_id_to_hue_index.get(cid, 0)
-            hue = int((hue_index * 360 / num_classes)) % 360
-            color = QColor.fromHsv(hue, 150, 100)
+
+            color = self._get_color_for_class(cid, saturation=180, value=200)
+
             item.setBackground(QBrush(color))
             class_table.setItem(row, 0, item)
+
         class_table.blockSignals(False)
 
     def update_class_filter_combo(self):
@@ -632,19 +635,35 @@ class MainWindow(QMainWindow):
         if item.column() != 1:
             return
         table = self.right_panel.segment_table
+        index_item = table.item(item.row(), 0)
+        if not index_item:
+            return
+
         table.blockSignals(True)
         try:
-            new_class_id = int(item.text())
-            original_index = table.item(item.row(), 0).data(Qt.ItemDataRole.UserRole)
+            new_class_id_text = item.text()
+            if not new_class_id_text.strip():
+                raise ValueError("Class ID cannot be empty.")
+            new_class_id = int(new_class_id_text)
+            original_index = index_item.data(Qt.ItemDataRole.UserRole)
+
+            if original_index is None or original_index >= len(self.segments):
+                raise IndexError("Invalid segment index found in table.")
+
             self.segments[original_index]["class_id"] = new_class_id
             if new_class_id >= self.next_class_id:
                 self.next_class_id = new_class_id + 1
             self.update_all_lists()
-        except (ValueError, TypeError):
-            original_index = table.item(item.row(), 0).data(Qt.ItemDataRole.UserRole)
-            item.setText(str(self.segments[original_index]["class_id"]))
-        table.blockSignals(False)
-        self.viewer.setFocus()
+        except (ValueError, TypeError, AttributeError, IndexError) as e:
+            original_index = index_item.data(Qt.ItemDataRole.UserRole)
+            if original_index is not None and original_index < len(self.segments):
+                original_class_id = self.segments[original_index].get("class_id")
+                item.setText(
+                    str(original_class_id) if original_class_id is not None else "N/A"
+                )
+        finally:
+            table.blockSignals(False)
+            self.viewer.setFocus()
 
     def get_selected_segment_indices(self):
         table = self.right_panel.segment_table
@@ -854,7 +873,7 @@ class MainWindow(QMainWindow):
 def main():
     app = QApplication(sys.argv)
     qdarktheme.setup_theme()
-    sam_model = SamModel(model_type="vit_h")  # one-time check/download
+    sam_model = SamModel(model_type="vit_h")
     main_win = MainWindow(sam_model)
     main_win.show()
     sys.exit(app.exec())

{lazylabel_gui-1.0.2 → lazylabel_gui-1.0.4}/src/lazylabel_gui.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: lazylabel-gui
-Version: 1.0.2
+Version: 1.0.4
 Summary: An image segmentation GUI for generating mask tensors.
 Author-email: "Deniz N. Cakan" <deniz.n.cakan@gmail.com>
 License: MIT License
@@ -70,11 +70,11 @@ Inspired by [LabelMe](https://github.com/wkentaro/labelme?tab=readme-ov-file#ins
 ## 🚀 Getting Started
 
 ### Prerequisites
-Ensure you have **Python 3.10** or newer.
+**Python 3.10**
 
 ### Installation
 
-#### For Users (via PyPI)
+#### For Users [via PyPI](https://pypi.org/project/lazylabel-gui/)
 1.  Install LazyLabel directly:
     ```bash
     pip install lazylabel-gui
@@ -87,7 +87,7 @@ Ensure you have **Python 3.10** or newer.
 #### For Developers (from Source)
 1.  Clone the repository:
     ```bash
-    git clone [https://github.com/dnzckn/LazyLabel.git](https://github.com/dnzckn/LazyLabel.git)
+    git clone https://github.com/dnzckn/LazyLabel.git
     cd LazyLabel
     ```
 2.  Install in editable mode, which links the installed package to your source directory: