lazylabel-gui 1.1.7__tar.gz → 1.1.8__tar.gz

{lazylabel_gui-1.1.7/src/lazylabel_gui.egg-info → lazylabel_gui-1.1.8}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: lazylabel-gui
-Version: 1.1.7
+Version: 1.1.8
 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

{lazylabel_gui-1.1.7 → lazylabel_gui-1.1.8}/pyproject.toml

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

{lazylabel_gui-1.1.7 → lazylabel_gui-1.1.8}/src/lazylabel/core/file_manager.py

@@ -17,12 +17,21 @@ class FileManager:
         self.segment_manager = segment_manager
 
     def save_npz(
-        self, image_path: str, image_size: tuple[int, int], class_order: list[int]
+        self,
+        image_path: str,
+        image_size: tuple[int, int],
+        class_order: list[int],
+        crop_coords: tuple[int, int, int, int] | None = None,
     ) -> str:
         """Save segments as NPZ file."""
         final_mask_tensor = self.segment_manager.create_final_mask_tensor(
             image_size, class_order
         )
+
+        # Apply crop if coordinates are provided
+        if crop_coords:
+            final_mask_tensor = self._apply_crop_to_mask(final_mask_tensor, crop_coords)
+
         npz_path = os.path.splitext(image_path)[0] + ".npz"
         np.savez_compressed(npz_path, mask=final_mask_tensor.astype(np.uint8))
         return npz_path
@@ -33,11 +42,16 @@ class FileManager:
         image_size: tuple[int, int],
         class_order: list[int],
         class_labels: list[str],
+        crop_coords: tuple[int, int, int, int] | None = None,
     ) -> str | None:
         """Save segments as YOLO format TXT file."""
         final_mask_tensor = self.segment_manager.create_final_mask_tensor(
             image_size, class_order
         )
+
+        # Apply crop if coordinates are provided
+        if crop_coords:
+            final_mask_tensor = self._apply_crop_to_mask(final_mask_tensor, crop_coords)
         output_path = os.path.splitext(image_path)[0] + ".txt"
         h, w = image_size
 
@@ -118,6 +132,35 @@ class FileManager:
                                 }
                             )
 
+    def _apply_crop_to_mask(
+        self, mask_tensor: np.ndarray, crop_coords: tuple[int, int, int, int]
+    ) -> np.ndarray:
+        """Apply crop to mask tensor by setting areas outside crop to 0."""
+        x1, y1, x2, y2 = crop_coords
+        h, w = mask_tensor.shape[:2]
+
+        # Create a copy of the mask tensor
+        cropped_mask = mask_tensor.copy()
+
+        # Set areas outside crop to 0
+        # Top area (0, 0, w, y1)
+        if y1 > 0:
+            cropped_mask[:y1, :, :] = 0
+
+        # Bottom area (0, y2, w, h)
+        if y2 < h:
+            cropped_mask[y2:, :, :] = 0
+
+        # Left area (0, y1, x1, y2)
+        if x1 > 0:
+            cropped_mask[y1:y2, :x1, :] = 0
+
+        # Right area (x2, y1, w, y2)
+        if x2 < w:
+            cropped_mask[y1:y2, x2:, :] = 0
+
+        return cropped_mask
+
     def is_image_file(self, filepath: str) -> bool:
         """Check if file is a supported image format."""
         return filepath.lower().endswith((".png", ".jpg", ".jpeg", ".tiff", ".tif"))

{lazylabel_gui-1.1.7 → lazylabel_gui-1.1.8}/src/lazylabel/core/model_manager.py

@@ -94,3 +94,15 @@ class ModelManager:
         return self.sam_model is not None and getattr(
             self.sam_model, "is_loaded", False
         )
+
+    def set_image_from_path(self, image_path: str) -> bool:
+        """Set image for SAM model from file path."""
+        if not self.is_model_available():
+            return False
+        return self.sam_model.set_image_from_path(image_path)
+
+    def set_image_from_array(self, image_array) -> bool:
+        """Set image for SAM model from numpy array."""
+        if not self.is_model_available():
+            return False
+        return self.sam_model.set_image_from_array(image_array)