imagebaker 0.0.49__py3-none-any.whl → 0.0.51__py3-none-any.whl

imagebaker/utils/vis.py

@@ -0,0 +1,174 @@
+from imagebaker.core.defs import PredictionResult
+
+import cv2
+import numpy as np
+from typing import List
+
+
+def annotate_detection(
+    image: np.ndarray,
+    results: List[PredictionResult],
+    color_map: dict[str, tuple[int, int, int]],
+    box_thickness: int = 2,
+    font_face: int = cv2.FONT_HERSHEY_SIMPLEX,
+    text_scale: float = 0.5,
+    text_thickness: int = 1,
+) -> np.ndarray:
+    """
+    Draw bounding boxes and labels on the image
+
+    Args:
+        image: The original image as a numpy array
+        results: List of PredictionResult objects
+
+    Returns:
+        Annotated image as a numpy array
+    """
+    annotated_image = image.copy()
+
+    for result in results:
+        # Extract data from result
+        box = result.rectangle  # [x1, y1, x2, y2]
+        score = result.score
+        class_name = result.class_name
+
+        if not box:
+            continue
+
+        # Get color for this class
+        color = color_map.get(
+            result.class_name, (0, 255, 0)
+        )  # Default to green if not found
+
+        # Draw bounding box
+        cv2.rectangle(
+            annotated_image,
+            (box[0], box[1]),
+            (box[2], box[3]),
+            color,
+            box_thickness,
+        )
+
+        # Prepare label text with class name and score
+        label_text = f"{class_name}: {score:.2f}"
+
+        # Calculate text size to create background rectangle
+        (text_width, text_height), baseline = cv2.getTextSize(
+            label_text,
+            font_face,
+            text_scale,
+            text_thickness,
+        )
+
+        # Draw text background
+        cv2.rectangle(
+            annotated_image,
+            (box[0], box[1] - text_height - 5),
+            (box[0] + text_width, box[1]),
+            color,
+            -1,  # Fill the rectangle
+        )
+
+        # Draw text
+        cv2.putText(
+            annotated_image,
+            label_text,
+            (box[0], box[1] - 5),
+            font_face,
+            text_scale,
+            (255, 255, 255),  # White text
+            text_thickness,
+        )
+
+    return annotated_image
+
+
+def annotate_segmentation(
+    image: np.ndarray,
+    results: List[PredictionResult],
+    color_map: dict[int, tuple[int, int, int]],
+    contour_thickness: int = 2,
+    mask_opacity: float = 0.5,
+    font_face: int = cv2.FONT_HERSHEY_SIMPLEX,
+    text_scale: float = 0.5,
+    text_thickness: int = 1,
+) -> np.ndarray:
+    """
+    Draw segmentation masks and contours on the image
+    """
+    annotated_image = image.copy()
+    mask_overlay = np.zeros_like(image)
+
+    for i, result in enumerate(results):
+        if (result.polygon is not None) or not result.mask:
+            continue
+
+        # Get color for this mask
+        color_idx = i % len(color_map)
+        color = color_map[color_idx]
+
+        # Create mask from polygons
+        mask = np.zeros((image.shape[0], image.shape[1]), dtype=np.uint8)
+        for poly in result.polygon:
+            poly_np = np.array(poly, dtype=np.int32).reshape((-1, 1, 2))
+            cv2.fillPoly(mask, [poly_np], 1)
+
+        # Apply color to mask overlay
+        color_mask = np.zeros_like(image)
+        color_mask[mask == 1] = color
+        mask_overlay = cv2.addWeighted(mask_overlay, 1.0, color_mask, 1.0, 0)
+
+        # Draw contours
+        for poly in result.polygon:
+            poly_np = np.array(poly, dtype=np.int32).reshape((-1, 1, 2))
+            cv2.polylines(
+                annotated_image,
+                [poly_np],
+                True,
+                color,
+                contour_thickness,
+            )
+
+        # Add label text
+        label_position = (
+            result.polygon[0][0] if result.polygon and result.polygon[0] else [10, 10]
+        )
+        label_text = f"{result.class_id}: {result.score:.2f}"
+
+        # Draw text background
+        (text_width, text_height), baseline = cv2.getTextSize(
+            label_text,
+            font_face,
+            text_scale,
+            text_thickness,
+        )
+
+        cv2.rectangle(
+            annotated_image,
+            (label_position[0], label_position[1] - text_height - 5),
+            (label_position[0] + text_width, label_position[1]),
+            color,
+            -1,  # Fill the rectangle
+        )
+
+        # Draw text
+        cv2.putText(
+            annotated_image,
+            label_text,
+            (label_position[0], label_position[1] - 5),
+            font_face,
+            text_scale,
+            (255, 255, 255),  # White text
+            text_thickness,
+        )
+
+    # Blend mask overlay with original image
+    annotated_image = cv2.addWeighted(
+        annotated_image,
+        1.0,
+        mask_overlay,
+        mask_opacity,
+        0,
+    )
+
+    return annotated_image

imagebaker/window/main_window.py

@@ -175,9 +175,15 @@ class MainWindow(QMainWindow):
         self.status_bar.showMessage(status_text)
 
     def closeEvent(self, event):
-        # Clean up tabs first
-        if hasattr(self, "layerify_tab"):
-            self.layerify_tab.deleteLater()
-        if hasattr(self, "baker_tab"):
-            self.baker_tab.deleteLater()
-        super().closeEvent(event)
+        logger.info("Closing the application.")
+        if self.layerify_config.cleanup_on_exit:
+            import shutil
+
+            if self.layerify_config.bake_dir.exists():
+                shutil.rmtree(self.layerify_config.bake_dir)
+                logger.info(f"Deleted bake directory: {self.layerify_config.bake_dir}")
+            if self.layerify_config.cache_dir.exists():
+                shutil.rmtree(self.layerify_config.cache_dir)
+                logger.info(
+                    f"Deleted cache directory: {self.layerify_config.cache_dir}"
+                )

imagebaker/workers/baker_worker.py

@@ -49,10 +49,23 @@ class BakerWorker(QObject):
                 top_left = QPointF(sys.maxsize, sys.maxsize)
                 bottom_right = QPointF(-sys.maxsize, -sys.maxsize)
 
+                # contains all states in currenct step
                 for state in states:
                     layer = self._get_layer(state.layer_id)
                     if layer and layer.visible and not layer.image.isNull():
+                        update_opacities = False
+                        logger.debug(
+                            f"Updating layer {layer.layer_name} with state: {state}"
+                        )
+
+                        if (
+                            layer.edge_width != state.edge_width
+                            or layer.edge_opacity != state.edge_opacity
+                        ):
+                            update_opacities = True
                         layer.layer_state = state
+                        if update_opacities:
+                            layer._apply_edge_opacity()
                         layer.update()
 
                         transform = QTransform()
@@ -211,6 +224,8 @@ class BakerWorker(QObject):
             self.finished.emit(results)
 
         except Exception as e:
+            import traceback
+
             logger.error(f"Error in BakerWorker: {e}")
             self.error.emit(str(e))
             traceback.print_exc()
@@ -235,9 +250,14 @@ class BakerWorker(QObject):
             new_annotation.points = ann.points
         elif ann.rectangle:
             xywhs = mask_to_rectangles(alpha_channel, merge_rectangles=True)
-            new_annotation.rectangle = QRectF(
-                xywhs[0][0], xywhs[0][1], xywhs[0][2], xywhs[0][3]
-            )
+            if len(xywhs) == 0:
+                logger.info("No rectangles found")
+                # return None
+            else:
+                logger.info(f"Found {len(xywhs)} rectangles")
+                new_annotation.rectangle = QRectF(
+                    xywhs[0][0], xywhs[0][1], xywhs[0][2], xywhs[0][3]
+                )
         elif ann.polygon:
             polygon = mask_to_polygons(alpha_channel, merge_polygons=True)
             poly = QPolygonF([QPointF(p[0], p[1]) for p in polygon[0]])

{imagebaker-0.0.49.dist-info → imagebaker-0.0.51.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: imagebaker
-Version: 0.0.49
+Version: 0.0.51
 Summary: A package for baking images.
 Home-page: https://github.com/q-viper/Image-Baker
 Author: Ramkrishna Acharya
@@ -35,7 +35,7 @@ Requires-Dist: mkdocs-awesome-pages-plugin; extra == "docs"
 ![code size in bytes](https://img.shields.io/github/languages/code-size/q-viper/image-baker)
 <!-- ![Tests](https://github.com/q-viper/SmokeSim/actions/workflows/test-on-push.yml/badge.svg) -->
 ![Code Formatting](https://github.com/q-viper/image-baker/actions/workflows/black-formatter.yml/badge.svg)
-
+[![PyPI version](https://img.shields.io/pypi/v/imagebaker.svg)](https://pypi.org/imagebaker/)
 
 <p align="center">
     <img src="assets/demo.gif" alt="Centered Demo" />
@@ -112,6 +112,9 @@ After cloning and going to the project directory, the following code should work
 * **Ctrl + D**: Draw Mode on Baker Tab. Drawing can happen on a selected or main layer.
 * **Ctrl + E**: Erase Mode on Baker Tab.
 * **Wheel**: Change the size of the drawing pointer.
+* **Q**: Point mode on annotation.
+* **W**: Polygon mode on annotation.
+* **E**: Rectangle mode on annotation.
 
 ## Demo
 ### Annotation Page

imagebaker-0.0.51.dist-info/RECORD

@@ -0,0 +1,43 @@
+imagebaker/__init__.py,sha256=zrrxwyzuqVNeIu3rVPrOGYf6SCd5kWsoGcdqqUfsYX4,258
+imagebaker/core/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+imagebaker/core/configs/__init__.py,sha256=iyR_GOVMFw3XJSm7293YfyTnaLZa7pLQMfn5tGxVofI,31
+imagebaker/core/configs/configs.py,sha256=hm0CHfMVTD9353wB61Df9bVUb898zQVIFjdoXELdJcc,5694
+imagebaker/core/defs/__init__.py,sha256=NqV7gYIlRkaS7nx_UTNPSNZbdPrx4w-VurKOKyRLbKY,28
+imagebaker/core/defs/defs.py,sha256=oZCkgqHgvE_yupUzD5IB_ZMoRhyTh0EdCR8Gmh6TMsI,8505
+imagebaker/core/plugins/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+imagebaker/core/plugins/base_plugin.py,sha256=ROa1HTwV5LgGL-40CHKk_5MZYI5QAT1MzpYO7Fx-9P0,1084
+imagebaker/core/plugins/cosine_plugin.py,sha256=IXBfvaoxrvf-hytg_dT1zFmOfDbcWXBZ7NvIFPJ2tWQ,1251
+imagebaker/layers/__init__.py,sha256=q1kUDHhUXEGBOdu6CHDfqCnE2mraLHRqh0DFHYTbnRY,158
+imagebaker/layers/annotable_layer.py,sha256=xkV6KTXN8-N6Wu3QAj5nQDMPZ4bRr673_fu5QuRKBj8,37128
+imagebaker/layers/base_layer.py,sha256=1K7Nt6OPITrILj-p4I6Jf0eaesCpdeecOXGj_8oAQb8,30650
+imagebaker/layers/canvas_layer.py,sha256=6quVe_Ieyz14zPL92xTfrinAT7X4yb9iyEl-f9SnQZU,42116
+imagebaker/list_views/__init__.py,sha256=Aa9slE6do8eYgZp77wrofpd_mlBDwxgF3adMyHYFanE,144
+imagebaker/list_views/annotation_list.py,sha256=w6jNMyo3nxhQKZx8MyusNvxs_R6Pj-b-YBI5e0APZVM,7528
+imagebaker/list_views/canvas_list.py,sha256=JYSYR0peGyJFJ6amL1894KsUHETPUkR3qAWdGL50Lbc,6717
+imagebaker/list_views/image_list.py,sha256=EX2YmoTRffBXTbbVnuflbdsxZQD8fgM2StSxg1a9ANA,5389
+imagebaker/list_views/layer_list.py,sha256=fLx3Ry72fas1W5y_V84hSp41ARneogQN3qjfYTOcpxY,14476
+imagebaker/list_views/layer_settings.py,sha256=0WVSCm_RSBKo4pCkYU5c2OYjb_sW8x0UUfFC4So26jQ,9752
+imagebaker/models/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+imagebaker/models/base_model.py,sha256=4RyS4vShqWFHhdQDwYluwTPnRPEXjpZl9UjYY_w8NL0,4203
+imagebaker/tabs/__init__.py,sha256=ijg7MA17RvcHA2AuZE4OgRJXWxjecaUAlfASKAoCQ6Q,86
+imagebaker/tabs/baker_tab.py,sha256=yFQRiNmLIua5BvgW7Ysj5FrNbTLhGxqGRTRCABxeTtw,20493
+imagebaker/tabs/layerify_tab.py,sha256=F3BK5XsVCteB-2zyFBKqeWyHgIif5cU3TBEMbuDH3QM,37415
+imagebaker/utils/__init__.py,sha256=I1z5VVEf6QPOMvVkgVHDauQ9ew7tcTVguV4Kdi3Lk4Y,130
+imagebaker/utils/image.py,sha256=2-wbwD3PMwefgswge0drFM1XfXE7yQ64lqZZ5PwyCWs,3165
+imagebaker/utils/state_utils.py,sha256=Y2JVRGVfsoffwfA2lsCcqHwIxH_jOrEJAdH-oWfe2XE,3841
+imagebaker/utils/transform_mask.py,sha256=k8MfTgM5-_U2TvDHQHRelz-leGFX6OcsllV6-J4BKfw,3651
+imagebaker/utils/utils.py,sha256=MnJ4flxxwZbjROWJ5iKHnJxPSSMbfWRbF9GKfVcKutA,840
+imagebaker/utils/vis.py,sha256=f7c44gm6g9ja5hgVeXKfOhHzxHdzXcIUwKiA1RZU_F8,4736
+imagebaker/window/__init__.py,sha256=FIxtUR1qnbQMYzppQv7tEfv1-ueHhpu0Z7xuWZR794w,44
+imagebaker/window/app.py,sha256=e6FGO_BnvkiQC9JN3AmqkgbF72zzZS0hc7PFc43QiVc,4725
+imagebaker/window/main_window.py,sha256=-CP7q6xzkhv9Cl5RwUft1Rv8nIAuj_SLOiZDS2x6QJ4,7374
+imagebaker/workers/__init__.py,sha256=XfXENwAYyNg9q_zR-gOsYJGjzwg_iIb_gING8ydnp9c,154
+imagebaker/workers/baker_worker.py,sha256=_jWeyYAGoO2mfxXDn7fBm9tIA69OITewDVN0hSAt3Jc,11532
+imagebaker/workers/layerify_worker.py,sha256=EOqKvhdACtf3y5Ljy6M7MvddAjlZW5DNfBFMtNPD-us,3223
+imagebaker/workers/model_worker.py,sha256=Tlg6_D977iK-kuGCNdQY4OnGiP8QqWY7adpRNXZw4rA,1636
+imagebaker-0.0.51.dist-info/LICENSE,sha256=1vkysFPOnT7y4LsoFTv9YsopIrQvBc2l6vUOfv4KKLc,1082
+imagebaker-0.0.51.dist-info/METADATA,sha256=1DumEPL4LyKVeuV_tDwyepZ9PsreraeKSTEJBernVmw,6940
+imagebaker-0.0.51.dist-info/WHEEL,sha256=tZoeGjtWxWRfdplE7E3d45VPlLNQnvbKiYnx7gwAy8A,92
+imagebaker-0.0.51.dist-info/entry_points.txt,sha256=IDjZHJCiiHpH5IUTByT2en0nMbnnnlrJZ5FPFehUvQM,61
+imagebaker-0.0.51.dist-info/top_level.txt,sha256=Gg-eILTlqJXwVQr0saSwsx3-H4SPdZ2agBZaufe194s,11
+imagebaker-0.0.51.dist-info/RECORD,,

imagebaker-0.0.49.dist-info/RECORD

@@ -1,41 +0,0 @@
-imagebaker/__init__.py,sha256=ep22Ki-KljDvYzGvAfrDe9UU77GWgfoAsLd7t0S273c,130
-imagebaker/core/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-imagebaker/core/configs/__init__.py,sha256=iyR_GOVMFw3XJSm7293YfyTnaLZa7pLQMfn5tGxVofI,31
-imagebaker/core/configs/configs.py,sha256=uyUhoHA9-Ult-vVtnLX-JNmeUght4mUUV80yKWzUO_o,5100
-imagebaker/core/defs/__init__.py,sha256=NqV7gYIlRkaS7nx_UTNPSNZbdPrx4w-VurKOKyRLbKY,28
-imagebaker/core/defs/defs.py,sha256=nIg2ZQADbpcyC0ZOl54L14yLZ38-SUnfMkklCDfhN3E,8257
-imagebaker/core/plugins/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-imagebaker/core/plugins/base_plugin.py,sha256=ROa1HTwV5LgGL-40CHKk_5MZYI5QAT1MzpYO7Fx-9P0,1084
-imagebaker/core/plugins/cosine_plugin.py,sha256=IXBfvaoxrvf-hytg_dT1zFmOfDbcWXBZ7NvIFPJ2tWQ,1251
-imagebaker/layers/__init__.py,sha256=q1kUDHhUXEGBOdu6CHDfqCnE2mraLHRqh0DFHYTbnRY,158
-imagebaker/layers/annotable_layer.py,sha256=ZjDqNMEqB78iEAThotfksv8-xIK6NCr52fI26Rv6CXA,36627
-imagebaker/layers/base_layer.py,sha256=uncseSxTbLKnV84hAPbJLDbBMV0nu1gQ4N-SLHoGB6s,25486
-imagebaker/layers/canvas_layer.py,sha256=47R-g3NAzAQ720tWrIZUYIAupxImdORVYK5BFIRopqo,40414
-imagebaker/list_views/__init__.py,sha256=Aa9slE6do8eYgZp77wrofpd_mlBDwxgF3adMyHYFanE,144
-imagebaker/list_views/annotation_list.py,sha256=HGV6lGlkFjvJvvGnCcLuX1kkfXA0GL8wKo8jOXSBXec,7527
-imagebaker/list_views/canvas_list.py,sha256=JYSYR0peGyJFJ6amL1894KsUHETPUkR3qAWdGL50Lbc,6717
-imagebaker/list_views/image_list.py,sha256=NInkc893FGU7L6oSxy8KrWql-i6RB0BqvkroPAASjVw,4912
-imagebaker/list_views/layer_list.py,sha256=fLx3Ry72fas1W5y_V84hSp41ARneogQN3qjfYTOcpxY,14476
-imagebaker/list_views/layer_settings.py,sha256=38E39z-rEdl0YSe2C_k4wd5CgHPETQeJE5VvJLfFQ-k,8454
-imagebaker/models/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-imagebaker/models/base_model.py,sha256=ZfZ_VgP-jzfmEAdOycVtOYkr03m8EyXWdmK0-50MxIk,4197
-imagebaker/tabs/__init__.py,sha256=ijg7MA17RvcHA2AuZE4OgRJXWxjecaUAlfASKAoCQ6Q,86
-imagebaker/tabs/baker_tab.py,sha256=kIVoF4w_Ch1YNdHtHsNZKfr46tZWBDSj5xrrEkF1piM,20752
-imagebaker/tabs/layerify_tab.py,sha256=2IEdRyYHhNdY2696YiLcnWWKYN6gYwcKH6jsffYiL5M,32304
-imagebaker/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-imagebaker/utils/image.py,sha256=fq7g3DqSdjF9okxZ3fe5kF4Hxn32rqhvVqxy8yI5bnI,3067
-imagebaker/utils/state_utils.py,sha256=sI0R3ht4W_kQGwOpzUdBjy6M4PaKaAQlj-_MM6pidfM,3584
-imagebaker/utils/transform_mask.py,sha256=k8MfTgM5-_U2TvDHQHRelz-leGFX6OcsllV6-J4BKfw,3651
-imagebaker/window/__init__.py,sha256=FIxtUR1qnbQMYzppQv7tEfv1-ueHhpu0Z7xuWZR794w,44
-imagebaker/window/app.py,sha256=e6FGO_BnvkiQC9JN3AmqkgbF72zzZS0hc7PFc43QiVc,4725
-imagebaker/window/main_window.py,sha256=gpJ7DDuPmxhHh_6Rv3YH2J_1AqG7-NM8R3tKNYhFT3E,7030
-imagebaker/workers/__init__.py,sha256=XfXENwAYyNg9q_zR-gOsYJGjzwg_iIb_gING8ydnp9c,154
-imagebaker/workers/baker_worker.py,sha256=JyV1Hu4mzbYhmogc7K3U24adklmT3x3V0ZNMxe7iT-w,10697
-imagebaker/workers/layerify_worker.py,sha256=EOqKvhdACtf3y5Ljy6M7MvddAjlZW5DNfBFMtNPD-us,3223
-imagebaker/workers/model_worker.py,sha256=Tlg6_D977iK-kuGCNdQY4OnGiP8QqWY7adpRNXZw4rA,1636
-imagebaker-0.0.49.dist-info/LICENSE,sha256=1vkysFPOnT7y4LsoFTv9YsopIrQvBc2l6vUOfv4KKLc,1082
-imagebaker-0.0.49.dist-info/METADATA,sha256=BldoEWej9PUXnJVw_rpJEYFnjXosJIXWKJB2oY9ZLyI,6736
-imagebaker-0.0.49.dist-info/WHEEL,sha256=tZoeGjtWxWRfdplE7E3d45VPlLNQnvbKiYnx7gwAy8A,92
-imagebaker-0.0.49.dist-info/entry_points.txt,sha256=IDjZHJCiiHpH5IUTByT2en0nMbnnnlrJZ5FPFehUvQM,61
-imagebaker-0.0.49.dist-info/top_level.txt,sha256=Gg-eILTlqJXwVQr0saSwsx3-H4SPdZ2agBZaufe194s,11
-imagebaker-0.0.49.dist-info/RECORD,,