lazylabel-gui 1.0.6__py3-none-any.whl → 1.0.7__py3-none-any.whl

lazylabel/photo_viewer.py

@@ -1,5 +1,5 @@
 from PyQt6.QtCore import Qt, QRectF
-from PyQt6.QtGui import QPixmap
+from PyQt6.QtGui import QPixmap, QCursor
 from PyQt6.QtWidgets import QGraphicsView, QGraphicsScene, QGraphicsPixmapItem
 
 
@@ -21,8 +21,7 @@ class PhotoViewer(QGraphicsView):
         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())
+            self.resetTransform()
             viewrect = self.viewport().rect()
             scenerect = self.transform().mapRect(rect)
             factor = min(
@@ -30,6 +29,7 @@ class PhotoViewer(QGraphicsView):
                 viewrect.height() / scenerect.height(),
             )
             self.scale(factor, factor)
+            self.centerOn(self._pixmap_item)
 
     def set_photo(self, pixmap):
         if pixmap and not pixmap.isNull():
@@ -38,6 +38,9 @@ class PhotoViewer(QGraphicsView):
         else:
             self._pixmap_item.setPixmap(QPixmap())
 
+    def set_cursor(self, cursor_shape):
+        self.viewport().setCursor(QCursor(cursor_shape))
+
     def resizeEvent(self, event):
         self.fitInView()
         super().resizeEvent(event)

lazylabel/reorderable_class_table.py

@@ -34,11 +34,13 @@ class ReorderableClassTable(QTableWidget):
                 list({index.row() for index in self.selectedIndexes()}), reverse=True
             )
 
-            dragged_items = []
+            dragged_rows_data = []
             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)
+                # Take all items from the row
+                row_data = [
+                    self.takeItem(row, col) for col in range(self.columnCount())
+                ]
+                dragged_rows_data.insert(0, row_data)
                 # Then remove the row itself
                 self.removeRow(row)
 
@@ -47,10 +49,11 @@ class ReorderableClassTable(QTableWidget):
                 if row < drop_row:
                     drop_row -= 1
 
-            # Insert items at the new location
-            for item in dragged_items:
+            # Insert rows and their items at the new location
+            for row_data in dragged_rows_data:
                 self.insertRow(drop_row)
-                self.setItem(drop_row, 0, item)
+                for col, item in enumerate(row_data):
+                    self.setItem(drop_row, col, item)
                 self.selectRow(drop_row)
                 drop_row += 1
 

lazylabel/utils.py

@@ -2,10 +2,10 @@ import numpy as np
 from PyQt6.QtGui import QImage, QPixmap
 
 
-def mask_to_pixmap(mask, color):
+def mask_to_pixmap(mask, color, alpha=150):
     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
+    colored_mask[mask, 3] = alpha  # Alpha channel for transparency
     image = QImage(
         colored_mask.data, mask.shape[1], mask.shape[0], QImage.Format.Format_RGBA8888
     )

{lazylabel_gui-1.0.6.dist-info → lazylabel_gui-1.0.7.dist-info}/METADATA

@@ -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.7.dist-info/RECORD

@@ -0,0 +1,17 @@
+lazylabel/controls.py,sha256=WfI0aIO1nfJ7YGTsEyP5Oc6Cd5A1Tbar943eaE_-KEY,6312
+lazylabel/custom_file_system_model.py,sha256=YSM1CN5bza7Q2Cjb4unXgvc-SbYwvhUQNS1_G7ncofk,2302
+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=TPpPIv74IetnwQbQh8pUtUey74KO3rISFXPBLJSlrtM,46445
+lazylabel/numeric_table_widget_item.py,sha256=ZnwaUvCeOGEX504DfbLHWKMKVMt5zSjdQkPjPCuYCcY,346
+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=3NfzgxTgYoevv82owmcGzLqrIBdE_B4DlWgumVf4y_E,448
+lazylabel_gui-1.0.7.dist-info/licenses/LICENSE,sha256=kSDEIgrWAPd1u2UFGGpC9X71dhzrlzBFs8hbDlENnGE,1092
+lazylabel_gui-1.0.7.dist-info/METADATA,sha256=MskYLxZx4sccjcAcf2RWH_4LLPho4Xz0w8W8f8z7iU4,6292
+lazylabel_gui-1.0.7.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+lazylabel_gui-1.0.7.dist-info/entry_points.txt,sha256=Hd0WwEG9OPTa_ziYjiD0aRh7R6Fupt-wdQ3sspdc1mM,54
+lazylabel_gui-1.0.7.dist-info/top_level.txt,sha256=YN4uIyrpDBq1wiJaBuZLDipIzyZY0jqJOmmXiPIOUkU,10
+lazylabel_gui-1.0.7.dist-info/RECORD,,

lazylabel_gui-1.0.6.dist-info/RECORD

@@ -1,16 +0,0 @@
-lazylabel/controls.py,sha256=YCTm8hQMgk4GFrr7b8p5p8PX3IQi9NoWZV1EFYEhDI8,4125
-lazylabel/custom_file_system_model.py,sha256=c1cxQqrfZ7d0Tz9kHpygiIEqw4ffuEBjkD8wjumNaSY,2005
-lazylabel/editable_vertex.py,sha256=lw_MmDmgkiPZbouIb6DkqIEJZhG32AJ2T2TUt0P3rWk,1040
-lazylabel/hoverable_polygon_item.py,sha256=-0l8C8PfsXtJGqvZZ2qtizxHmFwO8RCwz5UfjKpDvzY,775
-lazylabel/main.py,sha256=EUTmxS2AiJ7BEYkKIkCfhn2rraxZHX00kQ98eMTnxy0,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.6.dist-info/licenses/LICENSE,sha256=kSDEIgrWAPd1u2UFGGpC9X71dhzrlzBFs8hbDlENnGE,1092
-lazylabel_gui-1.0.6.dist-info/METADATA,sha256=uHTTXfYNtNuazxOxyWIdBwt9TRM0B4htdfH8LhgGyd4,6259
-lazylabel_gui-1.0.6.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-lazylabel_gui-1.0.6.dist-info/entry_points.txt,sha256=Hd0WwEG9OPTa_ziYjiD0aRh7R6Fupt-wdQ3sspdc1mM,54
-lazylabel_gui-1.0.6.dist-info/top_level.txt,sha256=YN4uIyrpDBq1wiJaBuZLDipIzyZY0jqJOmmXiPIOUkU,10
-lazylabel_gui-1.0.6.dist-info/RECORD,,