lazylabel-gui 1.0.0__py3-none-any.whl

lazylabel/numeric_table_widget_item.py

@@ -0,0 +1,10 @@
+from PyQt6.QtWidgets import QTableWidgetItem
+
+
+class NumericTableWidgetItem(QTableWidgetItem):
+    def __lt__(self, other):
+        # Override the default less-than operator for sorting
+        try:
+            return int(self.text()) < int(other.text())
+        except (ValueError, TypeError):
+            return super().__lt__(other)

lazylabel/photo_viewer.py

@@ -0,0 +1,51 @@
+from PyQt6.QtCore import Qt, QRectF
+from PyQt6.QtGui import QPixmap
+from PyQt6.QtWidgets import QGraphicsView, QGraphicsScene, QGraphicsPixmapItem
+
+
+class PhotoViewer(QGraphicsView):
+    def __init__(self, parent=None):
+        super().__init__(parent)
+        self._scene = QGraphicsScene(self)
+        self._pixmap_item = QGraphicsPixmapItem()
+        self._scene.addItem(self._pixmap_item)
+        self.setScene(self._scene)
+
+        self.setTransformationAnchor(QGraphicsView.ViewportAnchor.AnchorUnderMouse)
+        self.setResizeAnchor(QGraphicsView.ViewportAnchor.AnchorViewCenter)
+        self.setVerticalScrollBarPolicy(Qt.ScrollBarPolicy.ScrollBarAlwaysOff)
+        self.setHorizontalScrollBarPolicy(Qt.ScrollBarPolicy.ScrollBarAlwaysOff)
+        self.setDragMode(QGraphicsView.DragMode.NoDrag)
+
+    def fitInView(self, scale=True):
+        rect = QRectF(self._pixmap_item.pixmap().rect())
+        if not rect.isNull():
+            self.setSceneRect(rect)
+            unity = self.transform().mapRect(QRectF(0, 0, 1, 1))
+            self.scale(1 / unity.width(), 1 / unity.height())
+            viewrect = self.viewport().rect()
+            scenerect = self.transform().mapRect(rect)
+            factor = min(
+                viewrect.width() / scenerect.width(),
+                viewrect.height() / scenerect.height(),
+            )
+            self.scale(factor, factor)
+
+    def set_photo(self, pixmap):
+        if pixmap and not pixmap.isNull():
+            self._pixmap_item.setPixmap(pixmap)
+            self.fitInView()
+        else:
+            self._pixmap_item.setPixmap(QPixmap())
+
+    def resizeEvent(self, event):
+        self.fitInView()
+        super().resizeEvent(event)
+
+    def wheelEvent(self, event):
+        if not self._pixmap_item.pixmap().isNull():
+            if event.angleDelta().y() > 0:
+                factor = 1.25
+            else:
+                factor = 0.8
+            self.scale(factor, factor)

lazylabel/reorderable_class_table.py

@@ -0,0 +1,58 @@
+from PyQt6.QtWidgets import QTableWidget, QAbstractItemView
+from PyQt6.QtCore import Qt
+
+
+class ReorderableClassTable(QTableWidget):
+    def __init__(self, *args, **kwargs):
+        super().__init__(*args, **kwargs)
+        self.setDragEnabled(True)
+        self.setAcceptDrops(True)
+        self.setDropIndicatorShown(True)
+        self.setSelectionBehavior(QAbstractItemView.SelectionBehavior.SelectRows)
+        self.setSelectionMode(QAbstractItemView.SelectionMode.ExtendedSelection)
+        self.setDragDropMode(QAbstractItemView.DragDropMode.InternalMove)
+        self.scroll_margin = 40
+
+    def dragMoveEvent(self, event):
+        pos = event.position().toPoint()
+        rect = self.viewport().rect()
+
+        if pos.y() < rect.top() + self.scroll_margin:
+            self.verticalScrollBar().setValue(self.verticalScrollBar().value() - 1)
+        elif pos.y() > rect.bottom() - self.scroll_margin:
+            self.verticalScrollBar().setValue(self.verticalScrollBar().value() + 1)
+
+        super().dragMoveEvent(event)
+
+    def dropEvent(self, event):
+        if not event.isAccepted() and event.source() == self:
+            drop_row = self.rowAt(event.position().toPoint().y())
+            if drop_row < 0:
+                drop_row = self.rowCount()
+
+            selected_rows = sorted(
+                list({index.row() for index in self.selectedIndexes()}), reverse=True
+            )
+
+            dragged_items = []
+            for row in selected_rows:
+                # Take the item from the row and keep its data
+                item = self.takeItem(row, 0)
+                dragged_items.insert(0, item)
+                # Then remove the row itself
+                self.removeRow(row)
+
+            # Adjust drop row if it was shifted by the removal
+            for row in selected_rows:
+                if row < drop_row:
+                    drop_row -= 1
+
+            # Insert items at the new location
+            for item in dragged_items:
+                self.insertRow(drop_row)
+                self.setItem(drop_row, 0, item)
+                self.selectRow(drop_row)
+                drop_row += 1
+
+            event.accept()
+        super().dropEvent(event)

lazylabel/sam_model.py

@@ -0,0 +1,70 @@
+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/utils.py

@@ -0,0 +1,12 @@
+import numpy as np
+from PyQt6.QtGui import QImage, QPixmap
+
+
+def mask_to_pixmap(mask, color):
+    colored_mask = np.zeros((mask.shape[0], mask.shape[1], 4), dtype=np.uint8)
+    colored_mask[mask, :3] = color
+    colored_mask[mask, 3] = 150  # Alpha channel for transparency
+    image = QImage(
+        colored_mask.data, mask.shape[1], mask.shape[0], QImage.Format.Format_RGBA8888
+    )
+    return QPixmap.fromImage(image)

lazylabel_gui-1.0.0.dist-info/METADATA

@@ -0,0 +1,147 @@
+Metadata-Version: 2.4
+Name: lazylabel-gui
+Version: 1.0.0
+Summary: An image segmentation GUI for generating mask tensors.
+Author-email: "Deniz N. Cakan" <deniz.n.cakan@gmail.com>
+License: MIT License
+        
+        Copyright (c) 2025 Deniz N. Cakan
+        
+        Permission is hereby granted, free of charge, to any person obtaining a copy
+        of this software and associated documentation files (the "Software"), to deal
+        in the Software without restriction, including without limitation the rights
+        to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+        copies of the Software, and to permit persons to whom the Software is
+        furnished to do so, subject to the following conditions:
+        
+        The above copyright notice and this permission notice shall be included in all
+        copies or substantial portions of the Software.
+        
+        THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+        IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+        FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+        AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+        LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+        OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+        SOFTWARE.
+        
+Project-URL: Homepage, https://github.com/dnzckn/lazylabel
+Project-URL: Bug Tracker, https://github.com/dnzckn/lazylabel/issues
+Classifier: Programming Language :: Python :: 3
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Operating System :: OS Independent
+Classifier: Topic :: Scientific/Engineering :: Image Processing
+Classifier: Environment :: X11 Applications :: Qt
+Requires-Python: >=3.10
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: PyQt6>=6.9.0
+Requires-Dist: pyqtdarktheme==2.1.0
+Requires-Dist: torch>=2.7.1
+Requires-Dist: torchvision>=0.22.1
+Requires-Dist: segment-anything==1.0
+Requires-Dist: numpy>=2.3.0
+Requires-Dist: opencv-python>=4.11.0.86
+Requires-Dist: scipy>=1.15.3
+Requires-Dist: requests>=2.32.4
+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.
+
+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 Screenshot](https://raw.githubusercontent.com/dnzckn/LazyLabel/main/src/lazylabel/demo_pictures/gui.PNG)
+
+---
+
+## ✨ 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.
+
+---
+
+## 🚀 Getting Started
+
+### Prerequisites
+Ensure you have **Python 3.10** or newer.
+
+### Installation
+
+#### For Users (via PyPI)
+1.  Install LazyLabel directly:
+    ```bash
+    pip install lazylabel-gui
+    ```
+2.  Run the application:
+    ```bash
+    lazylabel-gui
+    ```
+
+#### For Developers (from Source)
+1.  Clone the repository:
+    ```bash
+    git clone [https://github.com/dnzckn/LazyLabel.git](https://github.com/dnzckn/LazyLabel.git)
+    cd LazyLabel
+    ```
+2.  Install in editable mode, which links the installed package to your source directory:
+    ```bash
+    pip install -e .
+    ```
+3.  Run the application:
+    ```bash
+    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.
+
+---
+
+## ⌨️ Controls & Keybinds
+
+### Modes
+| Key | Action |
+|---|---|
+| `1` | Enter **Point Mode** (for AI segmentation). |
+| `2` | Enter **Polygon Drawing Mode**. |
+| `E` | Toggle **Selection Mode** to select existing segments. |
+| `R` | Enter **Edit Mode** for selected polygons (drag shape or vertices). |
+| `Q` | Toggle **Pan Mode** (click and drag the image). |
+
+### Actions
+| Key(s) | Action |
+|---|---|
+| `L-Click` | Add positive point (Point Mode) or polygon vertex. |
+| `R-Click` | Add negative point (Point Mode). |
+| `Ctrl + Z` | Undo last point. |
+| `Spacebar` | Finalize and save current AI segment. |
+| `Enter` | **Save final mask for the current image to a `.npz` file.** |
+| `M` | **Merge** selected segments into a single class. |
+| `V` / `Delete` / `Backspace`| **Delete** selected segments. |
+| `C` | Clear temporary points/vertices. |
+| `W/A/S/D` | Pan image. |
+| `Scroll Wheel` | Zoom-in or -out. |
+
+---
+
+## 📦 Output Format
+
+LazyLabel saves your work as a compressed NumPy array (`.npz`) with the same name as your image file.
+
+The file contains a single data key, `'mask'`, holding a **one-hot encoded tensor** with the shape `(H, W, C)`:
+* `H`: Image height.
+* `W`: Image width.
+* `C`: Total unique classes.
+
+Each channel is a binary mask for a class, combining all assigned segments into a clean, ML-ready output.
+
+---
+
+## ☕ Support LazyLabel
+[If you found LazyLabel helpful, consider supporting the project!](https://buymeacoffee.com/dnzckn)

lazylabel_gui-1.0.0.dist-info/RECORD

@@ -0,0 +1,16 @@
+lazylabel/controls.py,sha256=YCTm8hQMgk4GFrr7b8p5p8PX3IQi9NoWZV1EFYEhDI8,4125
+lazylabel/custom_file_system_model.py,sha256=QwqlHKrvBZbpKLXhUdBbJMbdImp7j6MnI8JCs7yKOlw,1996
+lazylabel/editable_vertex.py,sha256=lw_MmDmgkiPZbouIb6DkqIEJZhG32AJ2T2TUt0P3rWk,1040
+lazylabel/hoverable_polygon_item.py,sha256=-0l8C8PfsXtJGqvZZ2qtizxHmFwO8RCwz5UfjKpDvzY,775
+lazylabel/main.py,sha256=mJxZcy8mXXs7F7GD9BKyo0oGOHUu5HXSP1uDdsGNUno,35234
+lazylabel/numeric_table_widget_item.py,sha256=ZnwaUvCeOGEX504DfbLHWKMKVMt5zSjdQkPjPCuYCcY,346
+lazylabel/photo_viewer.py,sha256=qZ6t9RO0TOXQuoLr6ASVy6sgSIcVFPwyNRnjkK5MIHc,1993
+lazylabel/reorderable_class_table.py,sha256=9BctudGTrb6dTV8_7IpQ5lMM0D-5unuQbLVk3Yu49xQ,2242
+lazylabel/sam_model.py,sha256=9NB51Xq1P5dIxZMBdttwwRlszlJR3U5HRs83QsPLxNE,2595
+lazylabel/utils.py,sha256=a2n2f4NehzSjS8-UwbOioYTSO_lWe1uYjzDdOIEJloE,435
+lazylabel_gui-1.0.0.dist-info/licenses/LICENSE,sha256=kSDEIgrWAPd1u2UFGGpC9X71dhzrlzBFs8hbDlENnGE,1092
+lazylabel_gui-1.0.0.dist-info/METADATA,sha256=WmK28eQsDWHa8lXwZn6594bIGCFVUNWKP4IkZvQmXBE,6287
+lazylabel_gui-1.0.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+lazylabel_gui-1.0.0.dist-info/entry_points.txt,sha256=Hd0WwEG9OPTa_ziYjiD0aRh7R6Fupt-wdQ3sspdc1mM,54
+lazylabel_gui-1.0.0.dist-info/top_level.txt,sha256=YN4uIyrpDBq1wiJaBuZLDipIzyZY0jqJOmmXiPIOUkU,10
+lazylabel_gui-1.0.0.dist-info/RECORD,,

lazylabel_gui-1.0.0.dist-info/WHEEL

@@ -0,0 +1,5 @@
+Wheel-Version: 1.0
+Generator: setuptools (80.9.0)
+Root-Is-Purelib: true
+Tag: py3-none-any
+

lazylabel_gui-1.0.0.dist-info/entry_points.txt

@@ -0,0 +1,2 @@
+[console_scripts]
+lazylabel-gui = lazylabel.main:main

lazylabel_gui-1.0.0.dist-info/licenses/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2025 Deniz N. Cakan
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

lazylabel_gui-1.0.0.dist-info/top_level.txt

@@ -0,0 +1 @@
+lazylabel