lazylabel-gui 1.1.4__py3-none-any.whl → 1.1.6__py3-none-any.whl

lazylabel/ui/widgets/settings_widget.py

@@ -1,113 +1,125 @@
-"""Settings widget for save options."""
-
-from PyQt6.QtCore import pyqtSignal
-from PyQt6.QtWidgets import QCheckBox, QGroupBox, QVBoxLayout, QWidget
-
-
-class SettingsWidget(QWidget):
-    """Widget for application settings."""
-
-    settings_changed = pyqtSignal()
-
-    def __init__(self, parent=None):
-        super().__init__(parent)
-        self._setup_ui()
-        self._connect_signals()
-
-    def _setup_ui(self):
-        """Setup the UI layout."""
-        group = QGroupBox("Settings")
-        layout = QVBoxLayout(group)
-
-        # Auto-save
-        self.chk_auto_save = QCheckBox("Auto-Save on Navigate")
-        self.chk_auto_save.setToolTip(
-            "Automatically save work when using arrow keys to change images."
-        )
-        self.chk_auto_save.setChecked(True)
-        layout.addWidget(self.chk_auto_save)
-
-        # Save NPZ
-        self.chk_save_npz = QCheckBox("Save .npz")
-        self.chk_save_npz.setChecked(True)
-        self.chk_save_npz.setToolTip(
-            "Save the final mask as a compressed NumPy NPZ file."
-        )
-        layout.addWidget(self.chk_save_npz)
-
-        # Save TXT
-        self.chk_save_txt = QCheckBox("Save .txt")
-        self.chk_save_txt.setChecked(True)
-        self.chk_save_txt.setToolTip(
-            "Save bounding box annotations in YOLO TXT format."
-        )
-        layout.addWidget(self.chk_save_txt)
-
-        # YOLO with aliases
-        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.\n"
-            "This is useful when a separate .yaml or .names file defines the classes."
-        )
-        self.chk_yolo_use_alias.setChecked(True)
-        layout.addWidget(self.chk_yolo_use_alias)
-
-        # Save class aliases
-        self.chk_save_class_aliases = QCheckBox("Save Class Aliases (.json)")
-        self.chk_save_class_aliases.setToolTip(
-            "Save class aliases to a companion JSON file."
-        )
-        self.chk_save_class_aliases.setChecked(False)
-        layout.addWidget(self.chk_save_class_aliases)
-
-        # Main layout
-        main_layout = QVBoxLayout(self)
-        main_layout.setContentsMargins(0, 0, 0, 0)
-        main_layout.addWidget(group)
-
-    def _connect_signals(self):
-        """Connect internal signals."""
-        self.chk_save_npz.stateChanged.connect(self._handle_save_checkbox_change)
-        self.chk_save_txt.stateChanged.connect(self._handle_save_checkbox_change)
-
-        # Connect all checkboxes to settings changed signal
-        for checkbox in [
-            self.chk_auto_save,
-            self.chk_save_npz,
-            self.chk_save_txt,
-            self.chk_yolo_use_alias,
-            self.chk_save_class_aliases,
-        ]:
-            checkbox.stateChanged.connect(self.settings_changed)
-
-    def _handle_save_checkbox_change(self):
-        """Ensure at least one save format is selected."""
-        is_npz_checked = self.chk_save_npz.isChecked()
-        is_txt_checked = self.chk_save_txt.isChecked()
-
-        if not is_npz_checked and not is_txt_checked:
-            sender = self.sender()
-            if sender == self.chk_save_npz:
-                self.chk_save_txt.setChecked(True)
-            else:
-                self.chk_save_npz.setChecked(True)
-
-    def get_settings(self):
-        """Get current settings as dictionary."""
-        return {
-            "auto_save": self.chk_auto_save.isChecked(),
-            "save_npz": self.chk_save_npz.isChecked(),
-            "save_txt": self.chk_save_txt.isChecked(),
-            "yolo_use_alias": self.chk_yolo_use_alias.isChecked(),
-            "save_class_aliases": self.chk_save_class_aliases.isChecked(),
-        }
-
-    def set_settings(self, settings):
-        """Set settings from dictionary."""
-        self.chk_auto_save.setChecked(settings.get("auto_save", True))
-        self.chk_save_npz.setChecked(settings.get("save_npz", True))
-        self.chk_save_txt.setChecked(settings.get("save_txt", True))
-        self.chk_yolo_use_alias.setChecked(settings.get("yolo_use_alias", True))
-        self.chk_save_class_aliases.setChecked(
-            settings.get("save_class_aliases", False)
-        )
+"""Settings widget for save options."""
+
+from PyQt6.QtCore import pyqtSignal
+from PyQt6.QtWidgets import QCheckBox, QGroupBox, QVBoxLayout, QWidget
+
+
+class SettingsWidget(QWidget):
+    """Widget for application settings."""
+
+    settings_changed = pyqtSignal()
+
+    def __init__(self, parent=None):
+        super().__init__(parent)
+        self._setup_ui()
+        self._connect_signals()
+
+    def _setup_ui(self):
+        """Setup the UI layout."""
+        group = QGroupBox("Settings")
+        layout = QVBoxLayout(group)
+
+        # Auto-save
+        self.chk_auto_save = QCheckBox("Auto-Save on Navigate")
+        self.chk_auto_save.setToolTip(
+            "Automatically save work when using arrow keys to change images."
+        )
+        self.chk_auto_save.setChecked(True)
+        layout.addWidget(self.chk_auto_save)
+
+        # Save NPZ
+        self.chk_save_npz = QCheckBox("Save .npz")
+        self.chk_save_npz.setChecked(True)
+        self.chk_save_npz.setToolTip(
+            "Save the final mask as a compressed NumPy NPZ file."
+        )
+        layout.addWidget(self.chk_save_npz)
+
+        # Save TXT
+        self.chk_save_txt = QCheckBox("Save .txt")
+        self.chk_save_txt.setChecked(True)
+        self.chk_save_txt.setToolTip(
+            "Save bounding box annotations in YOLO TXT format."
+        )
+        layout.addWidget(self.chk_save_txt)
+
+        # YOLO with aliases
+        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.\n"
+            "This is useful when a separate .yaml or .names file defines the classes."
+        )
+        self.chk_yolo_use_alias.setChecked(True)
+        layout.addWidget(self.chk_yolo_use_alias)
+
+        # Save class aliases
+        self.chk_save_class_aliases = QCheckBox("Save Class Aliases (.json)")
+        self.chk_save_class_aliases.setToolTip(
+            "Save class aliases to a companion JSON file."
+        )
+        self.chk_save_class_aliases.setChecked(False)
+        layout.addWidget(self.chk_save_class_aliases)
+
+        # Operate on View
+        self.chk_operate_on_view = QCheckBox("Operate On View")
+        self.chk_operate_on_view.setToolTip(
+            "If checked, SAM model will operate on the currently displayed (adjusted) image.\n"
+            "Otherwise, it operates on the original image."
+        )
+        self.chk_operate_on_view.setChecked(False)
+        layout.addWidget(self.chk_operate_on_view)
+
+        # Main layout
+        main_layout = QVBoxLayout(self)
+        main_layout.setContentsMargins(0, 0, 0, 0)
+        main_layout.addWidget(group)
+
+    def _connect_signals(self):
+        """Connect internal signals."""
+        self.chk_save_npz.stateChanged.connect(self._handle_save_checkbox_change)
+        self.chk_save_txt.stateChanged.connect(self._handle_save_checkbox_change)
+
+        # Connect all checkboxes to settings changed signal
+        for checkbox in [
+            self.chk_auto_save,
+            self.chk_save_npz,
+            self.chk_save_txt,
+            self.chk_yolo_use_alias,
+            self.chk_save_class_aliases,
+            self.chk_operate_on_view,
+        ]:
+            checkbox.stateChanged.connect(self.settings_changed)
+
+    def _handle_save_checkbox_change(self):
+        """Ensure at least one save format is selected."""
+        is_npz_checked = self.chk_save_npz.isChecked()
+        is_txt_checked = self.chk_save_txt.isChecked()
+
+        if not is_npz_checked and not is_txt_checked:
+            sender = self.sender()
+            if sender == self.chk_save_npz:
+                self.chk_save_txt.setChecked(True)
+            else:
+                self.chk_save_npz.setChecked(True)
+
+    def get_settings(self):
+        """Get current settings as dictionary."""
+        return {
+            "auto_save": self.chk_auto_save.isChecked(),
+            "save_npz": self.chk_save_npz.isChecked(),
+            "save_txt": self.chk_save_txt.isChecked(),
+            "yolo_use_alias": self.chk_yolo_use_alias.isChecked(),
+            "save_class_aliases": self.chk_save_class_aliases.isChecked(),
+            "operate_on_view": self.chk_operate_on_view.isChecked(),
+        }
+
+    def set_settings(self, settings):
+        """Set settings from dictionary."""
+        self.chk_auto_save.setChecked(settings.get("auto_save", True))
+        self.chk_save_npz.setChecked(settings.get("save_npz", True))
+        self.chk_save_txt.setChecked(settings.get("save_txt", True))
+        self.chk_yolo_use_alias.setChecked(settings.get("yolo_use_alias", True))
+        self.chk_save_class_aliases.setChecked(
+            settings.get("save_class_aliases", False)
+        )
+        self.chk_operate_on_view.setChecked(settings.get("operate_on_view", False))

lazylabel/utils/logger.py

@@ -0,0 +1,45 @@
+import logging
+import os
+
+
+def setup_logging(log_file="lazylabel.log", level=logging.INFO):
+    """
+    Sets up the logging configuration for the application.
+
+    Args:
+        log_file (str): The name of the log file.
+        level (int): The logging level (e.g., logging.INFO, logging.DEBUG).
+    """
+    log_dir = os.path.join(os.path.expanduser("~"), ".lazylabel", "logs")
+    os.makedirs(log_dir, exist_ok=True)
+    log_path = os.path.join(log_dir, log_file)
+
+    # Create a logger
+    logger = logging.getLogger("lazylabel")
+    logger.setLevel(level)
+
+    # Create handlers
+    # Console handler
+    c_handler = logging.StreamHandler()
+    c_handler.setLevel(level)
+
+    # File handler
+    f_handler = logging.FileHandler(log_path)
+    f_handler.setLevel(level)
+
+    # Create formatters and add it to handlers
+    c_format = logging.Formatter("%(levelname)s: %(message)s")
+    f_format = logging.Formatter("%(asctime)s - %(name)s - %(levelname)s - %(message)s")
+    c_handler.setFormatter(c_format)
+    f_handler.setFormatter(f_format)
+
+    # Add handlers to the logger
+    if not logger.handlers:  # Avoid adding handlers multiple times
+        logger.addHandler(c_handler)
+        logger.addHandler(f_handler)
+
+    return logger
+
+
+# Initialize logger for the application
+logger = setup_logging()

{lazylabel_gui-1.1.4.dist-info → lazylabel_gui-1.1.6.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: lazylabel-gui
-Version: 1.1.4
+Version: 1.1.6
 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
@@ -139,6 +139,7 @@ Inspired by [LabelMe](https://github.com/wkentaro/labelme?tab=readme-ov-file#ins
 |---|---|
 | `1` | Enter **Point Mode** (for AI segmentation). |
 | `2` | Enter **Polygon Drawing Mode**. |
+| `3` | Enter **Bounding Box 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). |

{lazylabel_gui-1.1.4.dist-info → lazylabel_gui-1.1.6.dist-info}/RECORD

@@ -1,37 +1,39 @@
 lazylabel/__init__.py,sha256=0yitHZYNNuF4Rqj7cqE0TrfDV8zhzKD3A5W1vOymHK4,199
-lazylabel/main.py,sha256=AQ-SKj8korOa2rKviFaA1kC-c03VPmVN2xU2qk36764,768
+lazylabel/__main__.py,sha256=5IWklQrNkzeMH6rchzZxxoeSnZlkz-HRiPWXqjjj2yA,139
+lazylabel/main.py,sha256=8NkDYawiSRavXv35Lh2nYAz88EYVt8jrJXYZsk6EE7M,849
 lazylabel/config/__init__.py,sha256=TnDXH0yx0zy97FfjpiVPHleMgMrM5YwWrUbvevSNJjg,263
 lazylabel/config/hotkeys.py,sha256=bbHPpCNokwgH0EX74vGcNdn3RoyVZNVr-is6us-7J3M,7990
 lazylabel/config/paths.py,sha256=ZVKbtaNOxmYO4l6JgsY-8DXaE_jaJfDg2RQJJn3-5nw,1275
-lazylabel/config/settings.py,sha256=Ud27cSadJ7M4l30XN0KrxZ3ixAOPXTUhj-SvvanMjaI,1787
+lazylabel/config/settings.py,sha256=0muOT64zBr9Tn-JbyYRPf9xX760-X84ov93m56SpSHA,1898
 lazylabel/core/__init__.py,sha256=FmRjop_uIBSJwKMGhaZ-3Iwu34LkoxTuD-hnq5vbTSY,232
-lazylabel/core/file_manager.py,sha256=Blo55jXW-uGVLrreGRmlHUtK2yTbyj3jsTKkfjD0z7k,4796
-lazylabel/core/model_manager.py,sha256=P3IahI0xUk6-F7Y5U4r5RpnUQSelJqX2KBtcmUGNeNY,3368
-lazylabel/core/segment_manager.py,sha256=1lABdVdaa69_RPKx6ftYKJh7lNWCanodTsh7Utrpn2Q,6477
+lazylabel/core/file_manager.py,sha256=I8josIoPtRYQXcaBZbC5fb5zqCxbFlTVitkbKjyOz70,4715
+lazylabel/core/model_manager.py,sha256=PZj8YPLOacn_TnFcnPDY3stqapsT4B9mUYqJ500gCtM,3326
+lazylabel/core/segment_manager.py,sha256=M6kHcYeiub3WqL01NElCvKOc2GNmf72LUM1W8XwSaxc,6465
 lazylabel/models/__init__.py,sha256=fIlk_0DuZfiClcm0XlZdimeHzunQwBmTMI4PcGsaymw,91
-lazylabel/models/sam_model.py,sha256=89v99hpD0ngAAQKiuRyIib0E-5u9zTDJcubtmxxG-PM,7878
+lazylabel/models/sam_model.py,sha256=anh4XMkoX8U5MCZHV5HqN0x8iVnRYi50rcXFt6LdFCQ,8020
 lazylabel/ui/__init__.py,sha256=4qDIh9y6tABPmD8MAMGZn_G7oSRyrcHt2HkjoWgbGH4,268
-lazylabel/ui/control_panel.py,sha256=UlETN2IZv1l2xiaWPQvdalaK_f9YQf9iiB1Jb_EEgfs,9260
-lazylabel/ui/editable_vertex.py,sha256=xzc5QdFJJXiy035C3HgR4ph-tuJu8245l-Ar7aBvVv8,2399
-lazylabel/ui/hotkey_dialog.py,sha256=ZlZoPvY822ke8YkbCy22o2RUjxtkXSoEOg8eVETIKpQ,15118
+lazylabel/ui/control_panel.py,sha256=ZYclv2G1-HR_pg-b-4heU8hsKfE35jggLvcmJ3clHZA,10471
+lazylabel/ui/editable_vertex.py,sha256=nAFC2UuFfbvMbGBbAiLWA77cS5-Hn3a08xe1_QLz2yk,2449
+lazylabel/ui/hotkey_dialog.py,sha256=U_B76HLOxWdWkfA4d2XgRUaZTJPAAE_m5fmwf7Rh-5Y,14743
 lazylabel/ui/hoverable_pixelmap_item.py,sha256=kJFOp7WXiyHpNf7l73TZjiob85jgP30b5MZvu_z5L3c,728
 lazylabel/ui/hoverable_polygon_item.py,sha256=aclUwd0P8H8xbcep6GwhnfaVs1zSkqeZKAL-xeDyMiU,1222
-lazylabel/ui/main_window.py,sha256=ta3nzb6MWG-eLexlGKBc3oKtlsF9HpI-q1SvlrA-W2o,78623
+lazylabel/ui/main_window.py,sha256=zuszmw41849OXjBwVBc0TnBeuU4HZOJsJEurrzb_qbI,83764
 lazylabel/ui/numeric_table_widget_item.py,sha256=dQUlIFu9syCxTGAHVIlmbgkI7aJ3f3wmDPBz1AGK9Bg,283
-lazylabel/ui/photo_viewer.py,sha256=4n2zSpFAnausCoIU-wwWDHDo6vsNp_RGIkhDyhuM2_A,1997
+lazylabel/ui/photo_viewer.py,sha256=f93Mn9ajR2CYakJbbhhHvD5blKrwiGq3ZYgro-k2npc,4217
 lazylabel/ui/reorderable_class_table.py,sha256=sxHhQre5O_MXLDFgKnw43QnvXXoqn5xRKMGitgO7muI,2371
 lazylabel/ui/right_panel.py,sha256=-PeXcu7Lr-xhZniBMvWLDPiFb_RAHYAcILyw8fPJs6I,13139
 lazylabel/ui/widgets/__init__.py,sha256=bYjLRTqWdi4hcPfSSXmXuT-0c5Aee7HnnQurv_k5bfY,314
-lazylabel/ui/widgets/adjustments_widget.py,sha256=xM43q1khpvRIHslPfI6bukzjx9J0mPgHUdNKCmp3rtM,3912
+lazylabel/ui/widgets/adjustments_widget.py,sha256=CJXtq19hfy12ggCA3_RMwETKNnVAQsn-rT_lIYxpAT4,13458
 lazylabel/ui/widgets/model_selection_widget.py,sha256=kMPaBMfdfnEVH-Be1d5OxLEuioO0c04FZT1Hr904axM,3497
-lazylabel/ui/widgets/settings_widget.py,sha256=2T7A5t1g9oIRx6KsxWk--Reuft9DYeqH1rRI9ME18HQ,4347
+lazylabel/ui/widgets/settings_widget.py,sha256=qxVCiTYMQee95pE_FkFUXMCsmuo85LYdKeQYaCTrRVc,4829
 lazylabel/ui/widgets/status_bar.py,sha256=wTbMQNEOBfmtNj8EVFZS_lxgaemu-CbRXeZzEQDaVz8,4014
 lazylabel/utils/__init__.py,sha256=V6IR5Gim-39HgM2NyTVT-n8gy3mjilCSFW9y0owN5nc,179
 lazylabel/utils/custom_file_system_model.py,sha256=-3EimlybvevH6bvqBE0qdFnLADVtayylmkntxPXK0Bk,4869
+lazylabel/utils/logger.py,sha256=R7z6ifgA-NY-9ZbLlNH0i19zzwXndJ_gkG2J1zpVEhg,1306
 lazylabel/utils/utils.py,sha256=sYSCoXL27OaLgOZaUkCAhgmKZ7YfhR3Cc5F8nDIa3Ig,414
-lazylabel_gui-1.1.4.dist-info/licenses/LICENSE,sha256=kSDEIgrWAPd1u2UFGGpC9X71dhzrlzBFs8hbDlENnGE,1092
-lazylabel_gui-1.1.4.dist-info/METADATA,sha256=hYVIAtM7vZNaJ8-0QOnFtNyaK3UpScgswNyJuaiJaFE,9506
-lazylabel_gui-1.1.4.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-lazylabel_gui-1.1.4.dist-info/entry_points.txt,sha256=Hd0WwEG9OPTa_ziYjiD0aRh7R6Fupt-wdQ3sspdc1mM,54
-lazylabel_gui-1.1.4.dist-info/top_level.txt,sha256=YN4uIyrpDBq1wiJaBuZLDipIzyZY0jqJOmmXiPIOUkU,10
-lazylabel_gui-1.1.4.dist-info/RECORD,,
+lazylabel_gui-1.1.6.dist-info/licenses/LICENSE,sha256=kSDEIgrWAPd1u2UFGGpC9X71dhzrlzBFs8hbDlENnGE,1092
+lazylabel_gui-1.1.6.dist-info/METADATA,sha256=6zAAWlFPc61dOus5FGTAeEp4AlwNnbUjQsmHHwSgATQ,9546
+lazylabel_gui-1.1.6.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+lazylabel_gui-1.1.6.dist-info/entry_points.txt,sha256=Hd0WwEG9OPTa_ziYjiD0aRh7R6Fupt-wdQ3sspdc1mM,54
+lazylabel_gui-1.1.6.dist-info/top_level.txt,sha256=YN4uIyrpDBq1wiJaBuZLDipIzyZY0jqJOmmXiPIOUkU,10
+lazylabel_gui-1.1.6.dist-info/RECORD,,