imagebaker 0.0.50__py3-none-any.whl → 0.0.53__py3-none-any.whl

imagebaker/workers/baker_worker.py

@@ -39,6 +39,8 @@ class BakerWorker(QObject):
         self.layers = layers
         self.filename = filename
 
+        # logger.info(f"Received States: {self.states}")
+
     def process(self):
         results = []
         try:
@@ -128,6 +130,9 @@ class BakerWorker(QObject):
                                 painter.restore()
 
                             # Draw the drawing states
+                            logger.debug(
+                                f"Drawing states for layer {layer.layer_name}: {state.drawing_states}"
+                            )
                             if state.drawing_states:
                                 painter.save()
                                 try:
@@ -147,6 +152,10 @@ class BakerWorker(QObject):
                                         painter.drawPoint(
                                             drawing_state.position - top_left
                                         )
+                                except Exception as e:
+                                    logger.error(
+                                        f"Error drawing state for layer {layer.layer_name}: {e}"
+                                    )
                                 finally:
                                     painter.restore()
 
@@ -201,6 +210,7 @@ class BakerWorker(QObject):
                                 new_annotation = self._generate_annotation(
                                     ann, alpha_channel
                                 )
+                                new_annotation.caption = layer.caption
                                 new_annotations.append(new_annotation)
                 finally:
                     painter.end()
@@ -224,6 +234,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()
@@ -242,15 +254,22 @@ class BakerWorker(QObject):
             annotation_id=ann.annotation_id,
             is_complete=True,
             visible=True,
+            caption=ann.caption,
+            is_model_generated=ann.is_model_generated,
         )
 
         if ann.points:
             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.50.dist-info → imagebaker-0.0.53.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: imagebaker
-Version: 0.0.50
+Version: 0.0.53
 Summary: A package for baking images.
 Home-page: https://github.com/q-viper/Image-Baker
 Author: Ramkrishna Acharya
@@ -38,7 +38,7 @@ Requires-Dist: mkdocs-awesome-pages-plugin; extra == "docs"
 [![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" />
+    <img src="https://github.com/q-viper/image-baker/blob/main/assets/demo.gif?raw=true" alt="Centered Demo" />
 </p>
 
 
@@ -76,27 +76,28 @@ Run the following command to launch the GUI:
 
 `imagebaker`
 
-By default, the above command will not run any models on the backend. So please take a look into the example of model definition at [examples/loaded_models.py](examples/loaded_models.py). Then we need to pass it as:
+By default, the above command will not run any models on the backend. So please take a look into the example of model definition at [examples/loaded_models.py](https://github.com/q-viper/image-baker/blob/main/examples/loaded_models.py). Then we need to pass it as:
 
 `imagebaker --models-file examples/loaded_models.py`
 
 For more options, please do: `imagebaker --help` It should give the following options.
 
-![](assets/demo/options.png)
+![](https://github.com/q-viper/image-baker/blob/main/assets/demo/options.png?raw=true)
 
 
-* **`--configs-file`** allows us to define custom configs. The custom configs have to inherit LayerConfig and CanvasConfig defined at [imagebaker/core/configs/configs.py](imagebaker/core/configs/configs.py). An example is available at [examples](examples/). 
+* **`--configs-file`** allows us to define custom configs. The custom configs have to inherit LayerConfig and CanvasConfig defined at [imagebaker/core/configs/configs.py](https://github.com/q-viper/image-baker/blob/main/imagebaker/core/configs/configs.py). An example is available at [examples](https://github.com/q-viper/image-baker/blob/main/examples/). 
 
 After cloning and going to the project directory, the following code should work.
 `imagebaker --models-file examples/loaded_models.py --configs-file examples/example_config.py`
 
 ## Features
 - **Annotating Images**: Load a folder of images and annotate them using bounding boxes or polygons.
-- **Model Testing**: Define models for detection, segmentation, and prompts (e.g., points or rectangles) by following the base model structure in [imagebaker/models/base_model.py](imagebaker/models/base_model.py). See [examples/loaded_models.py](examples/loaded_models.py) for a working example.
+- **Model Testing**: Define models for detection, segmentation, and prompts (e.g., points or rectangles) by following the base model structure in [imagebaker/models/base_model.py](https://github.com/q-viper/image-baker/blob/main/imagebaker/models/base_model.py). See [examples/loaded_models.py](https://github.com/q-viper/image-baker/blob/main/examples/loaded_models.py) for a working example.
 - **Layerifying**: Crop images based on annotations to create reusable layers. Each cropped image represents a single layer.
 - **Baking States**: Arrange layers to create image variations by dragging, rotating, adjusting opacity, and more. Save the state using the Save State button or Ctrl + S.
 - **Playing States**: Replay saved states, export them locally, or use them for further predictions.
 - **Exporting States**: Export the final annotated JSON and the baked multilayer image.
+- **Drawing On Layers**: First select a layer then draw upon it. Only selected layer will be drawn. And if no layers are selected, then the drawing will not be exported.
 
 ### Shortcuts
 * **Ctrl + C**: Copy selected annotation/layer.
@@ -105,34 +106,44 @@ After cloning and going to the project directory, the following code should work
 * **Left Click**: Select an annotation/layer on mouse position.
 * **Left Click + Drag**: Drag a selected annotation/layer.
 * **Double Left Click**: When using polygon annotation, completes the polygon.
-* **Right Click**: Deselect an annotation/layer. While annotating the polygon, undo the last point.
+* **Right Click**: Unselect an annotation/layer. While annotating the polygon, undo the last point.
 * **Ctrl + Mouse Wheel**: Zoom In/Out on the mouse position, i.e., resize the viewport.
 * **Ctrl + Drag**: If done on the background, the viewport is panned.
 * **Ctrl + S**: Save State on Baker Tab.
 * **Ctrl + D**: Draw Mode on Baker Tab. Drawing can happen on a selected or main layer.
 * **Ctrl + E**: Erase Mode on Baker Tab.
+* **Ctrl + H**: Opens a help window.
 * **Wheel**: Change the size of the drawing pointer.
+* **Q**: Point mode on annotation.
+* **W**: Polygon mode on annotation. Moves selected layer one step up in layer lists in baker.
+* **S**: Moves selected layer one step down in layer list in baker.
+* **E**: Rectangle mode on annotation.
+* **H**: Hides/un-hides selected annotation/layer.
+* **L**: Creates layer from an annotation. If any annotation selected, creates only its, else creates layers from all visible annotations.
+* **C**: If any annotation is selected, a input box for Caption is created. It can be edited on baker tab as well and is state aware.
+* **Numerics**: Selecting number 1, 2, till 9 selects label. If not available, asks for a new label.
+* **Escape**: Closes the application.
 
 ## Demo
 ### Annotation Page
 This is where the loading of the image folder and annotation, connection with the model running in the backend, and layerifying happen.
 
-![](assets/demo/annotation_page.png)
+![](https://github.com/q-viper/image-baker/blob/main/assets/demo/annotation_page.png?raw=True)
 
 ### Baker Page
 This is where the layer baking happens. And the extraction of the layers as well.
 
-![](assets/demo/baker_page.png)
+![](https://github.com/q-viper/image-baker/blob/main/assets/demo/baker_page.png?raw=True)
 
 An example of drawing:
 
-![](assets/demo/drawing.png)
+![](https://github.com/q-viper/image-baker/blob/main/assets/demo/drawing.png?raw=True)
 
 ### Annotated
 
 The JSON and the baked image will be exported to the local folder, and in debug mode, the annotations and the mask for each layer will be exported too.
 
-![](assets/demo/annotated_veg_smiley.png)
+![](https://github.com/q-viper/image-baker/blob/main/assets/demo/annotated_veg_smiley.png?raw=True)
 
 ### Demo Video
 
@@ -151,3 +162,5 @@ Click on the image above to play the video on YouTube.
 Contributions are welcome! 
 
 Do you find this project to be useful and are you looking for some features that are not implemented yet? Feel free to open issues or submit pull requests to improve the project.
+
+For more please visit [CONTRIBUTING](CONTRIBUTING).

{imagebaker-0.0.50.dist-info → imagebaker-0.0.53.dist-info}/RECORD

@@ -1,43 +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=-qy7vmYaaUk3bh49pwsBig8de_3Y2JWjTDeyWcGQods,5130
+imagebaker/core/configs/configs.py,sha256=hm0CHfMVTD9353wB61Df9bVUb898zQVIFjdoXELdJcc,5694
 imagebaker/core/defs/__init__.py,sha256=NqV7gYIlRkaS7nx_UTNPSNZbdPrx4w-VurKOKyRLbKY,28
-imagebaker/core/defs/defs.py,sha256=-ZItfJdWaK9yFSuFn2LmQ4ncqukAZ_hvYFgE44HUdIo,8394
+imagebaker/core/defs/defs.py,sha256=67L65v8zTKwcrgEgJUVXn3n4xhVuylO23GLvJ-qD1ZU,8710
 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=ejBp6nooLtHs8c_G6lOsTJmSwBT404F_dmPsSkjvdEQ,37167
-imagebaker/layers/base_layer.py,sha256=1K7Nt6OPITrILj-p4I6Jf0eaesCpdeecOXGj_8oAQb8,30650
-imagebaker/layers/canvas_layer.py,sha256=7eDo0UHXRRZ5BX3BciPX88JsumzmwVemtS6cBjU9qwM,42115
+imagebaker/layers/annotable_layer.py,sha256=v8sE73ewESh6sMzFZGrcYTWI-FY-J5-noiIr4YN_rXc,40696
+imagebaker/layers/base_layer.py,sha256=Tbu1D6M2jzK_pIns3ghbHa3NK-htV-klxGo13lR5hNk,30841
+imagebaker/layers/canvas_layer.py,sha256=9nwRzEo1Bp5KrtVYPFKXUVMCqkSW2qflgKBpI0V9FzI,43132
 imagebaker/list_views/__init__.py,sha256=Aa9slE6do8eYgZp77wrofpd_mlBDwxgF3adMyHYFanE,144
-imagebaker/list_views/annotation_list.py,sha256=HGV6lGlkFjvJvvGnCcLuX1kkfXA0GL8wKo8jOXSBXec,7527
+imagebaker/list_views/annotation_list.py,sha256=3AN9uv-RcKFrJidfdC8R2m_tIbvf4IPRYcea4fCB5j4,7604
 imagebaker/list_views/canvas_list.py,sha256=JYSYR0peGyJFJ6amL1894KsUHETPUkR3qAWdGL50Lbc,6717
-imagebaker/list_views/image_list.py,sha256=NInkc893FGU7L6oSxy8KrWql-i6RB0BqvkroPAASjVw,4912
+imagebaker/list_views/image_list.py,sha256=kS6IMPhsyYXHZvx9lrUgPLmbZgNoiLlbFJW6wS6icjU,5582
 imagebaker/list_views/layer_list.py,sha256=fLx3Ry72fas1W5y_V84hSp41ARneogQN3qjfYTOcpxY,14476
-imagebaker/list_views/layer_settings.py,sha256=0WVSCm_RSBKo4pCkYU5c2OYjb_sW8x0UUfFC4So26jQ,9752
+imagebaker/list_views/layer_settings.py,sha256=KT0B4yIjR9DQmekIDh_TS0gBB6OkjdNBIZ1HnCFdUto,11025
 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=Qsf9w81-43sB0LZy_vIB4wCzUhHHhoIiarZJvKRxeZ0,32340
+imagebaker/tabs/baker_tab.py,sha256=8V1t-cDCSotz6ZDv6o-6HA8H9eli5_0vVBPkkUP9_x4,23091
+imagebaker/tabs/layerify_tab.py,sha256=4kbWOgDszuYEFI6VOrzqFpwvBAxe_ihFayubul30lz8,42703
 imagebaker/utils/__init__.py,sha256=I1z5VVEf6QPOMvVkgVHDauQ9ew7tcTVguV4Kdi3Lk4Y,130
-imagebaker/utils/image.py,sha256=fq7g3DqSdjF9okxZ3fe5kF4Hxn32rqhvVqxy8yI5bnI,3067
-imagebaker/utils/state_utils.py,sha256=Y2JVRGVfsoffwfA2lsCcqHwIxH_jOrEJAdH-oWfe2XE,3841
+imagebaker/utils/image.py,sha256=2-wbwD3PMwefgswge0drFM1XfXE7yQ64lqZZ5PwyCWs,3165
+imagebaker/utils/state_utils.py,sha256=6ivXVvamuYHQ6LjfoRIAuiTiZ4-LxUad22fvCH0CBcE,3932
 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=gpJ7DDuPmxhHh_6Rv3YH2J_1AqG7-NM8R3tKNYhFT3E,7030
+imagebaker/window/main_window.py,sha256=lpfTffUOKZx6rN_9uP1OQ3A2GRdRKzQYcDn-GZxAuLg,11342
 imagebaker/workers/__init__.py,sha256=XfXENwAYyNg9q_zR-gOsYJGjzwg_iIb_gING8ydnp9c,154
-imagebaker/workers/baker_worker.py,sha256=EJTL4ln09NuntFpu0o-Hfk0vCtDxpKqJxJcmtgTnMwo,11297
+imagebaker/workers/baker_worker.py,sha256=Bg5gcO91Ip1Egvs_hy9reQrsGNssbS7_nn_oYsmX5mE,12164
 imagebaker/workers/layerify_worker.py,sha256=EOqKvhdACtf3y5Ljy6M7MvddAjlZW5DNfBFMtNPD-us,3223
 imagebaker/workers/model_worker.py,sha256=Tlg6_D977iK-kuGCNdQY4OnGiP8QqWY7adpRNXZw4rA,1636
-imagebaker-0.0.50.dist-info/LICENSE,sha256=1vkysFPOnT7y4LsoFTv9YsopIrQvBc2l6vUOfv4KKLc,1082
-imagebaker-0.0.50.dist-info/METADATA,sha256=Zsl37tIGiV9wrQ2zfiaubhmPbhOMucos2V_fF5mtbXU,6829
-imagebaker-0.0.50.dist-info/WHEEL,sha256=tZoeGjtWxWRfdplE7E3d45VPlLNQnvbKiYnx7gwAy8A,92
-imagebaker-0.0.50.dist-info/entry_points.txt,sha256=IDjZHJCiiHpH5IUTByT2en0nMbnnnlrJZ5FPFehUvQM,61
-imagebaker-0.0.50.dist-info/top_level.txt,sha256=Gg-eILTlqJXwVQr0saSwsx3-H4SPdZ2agBZaufe194s,11
-imagebaker-0.0.50.dist-info/RECORD,,
+imagebaker-0.0.53.dist-info/LICENSE,sha256=1vkysFPOnT7y4LsoFTv9YsopIrQvBc2l6vUOfv4KKLc,1082
+imagebaker-0.0.53.dist-info/METADATA,sha256=_PgfaiQ55tSgQ-kZyPA-qc35RlpfFFf_n1aB3WXLLgw,8382
+imagebaker-0.0.53.dist-info/WHEEL,sha256=tZoeGjtWxWRfdplE7E3d45VPlLNQnvbKiYnx7gwAy8A,92
+imagebaker-0.0.53.dist-info/entry_points.txt,sha256=IDjZHJCiiHpH5IUTByT2en0nMbnnnlrJZ5FPFehUvQM,61
+imagebaker-0.0.53.dist-info/top_level.txt,sha256=Gg-eILTlqJXwVQr0saSwsx3-H4SPdZ2agBZaufe194s,11
+imagebaker-0.0.53.dist-info/RECORD,,