lazylabel-gui 1.0.2__py3-none-any.whl → 1.0.3__py3-none-any.whl

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"
@@ -118,10 +116,16 @@ class MainWindow(QMainWindow):
         self.control_panel.btn_clear_points.clicked.connect(self.clear_all_points)
 
     def set_mode(self, mode_name, is_toggle=False):
+        # Clear selection if exiting selection mode
+        if self.mode == "selection" and mode_name != "selection":
+            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"]:
             self.previous_mode = self.mode
+
         self.mode = mode_name
         self.control_panel.mode_label.setText(
             f"Mode: {mode_name.replace('_', ' ').title()}"
@@ -179,7 +183,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():
@@ -429,6 +435,7 @@ class MainWindow(QMainWindow):
         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)
@@ -854,7 +861,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.dist-info → lazylabel_gui-1.0.3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: lazylabel-gui
-Version: 1.0.2
+Version: 1.0.3
 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.dist-info → lazylabel_gui-1.0.3.dist-info}/RECORD

@@ -2,15 +2,15 @@ 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/main.py,sha256=jFt8Ln-l2pdrXcgz-iuB6vuDfrkgwq1VF97NnDLVf6Y,35337
 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.2.dist-info/licenses/LICENSE,sha256=kSDEIgrWAPd1u2UFGGpC9X71dhzrlzBFs8hbDlENnGE,1092
-lazylabel_gui-1.0.2.dist-info/METADATA,sha256=-DX-OtgjAiEbr6jViCUlKddSGw-RRM_DwNAgCLqaInQ,6287
-lazylabel_gui-1.0.2.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-lazylabel_gui-1.0.2.dist-info/entry_points.txt,sha256=Hd0WwEG9OPTa_ziYjiD0aRh7R6Fupt-wdQ3sspdc1mM,54
-lazylabel_gui-1.0.2.dist-info/top_level.txt,sha256=YN4uIyrpDBq1wiJaBuZLDipIzyZY0jqJOmmXiPIOUkU,10
-lazylabel_gui-1.0.2.dist-info/RECORD,,
+lazylabel_gui-1.0.3.dist-info/licenses/LICENSE,sha256=kSDEIgrWAPd1u2UFGGpC9X71dhzrlzBFs8hbDlENnGE,1092
+lazylabel_gui-1.0.3.dist-info/METADATA,sha256=PNqWmfgbgnWqOZLDbfg4nXlqQiKqwwCt_bRf5d9F_vI,6259
+lazylabel_gui-1.0.3.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+lazylabel_gui-1.0.3.dist-info/entry_points.txt,sha256=Hd0WwEG9OPTa_ziYjiD0aRh7R6Fupt-wdQ3sspdc1mM,54
+lazylabel_gui-1.0.3.dist-info/top_level.txt,sha256=YN4uIyrpDBq1wiJaBuZLDipIzyZY0jqJOmmXiPIOUkU,10
+lazylabel_gui-1.0.3.dist-info/RECORD,,