Add imageHeight and imageWidth in ai-annotate-rectangles

labelmeai · Oct 14, 2024 · 2c06965 · 2c06965
1 parent ba705b7
commit 2c06965
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/labelme_toolkit/_cli/_ai_annotate_rectangles.py b/labelme_toolkit/_cli/_ai_annotate_rectangles.py
@@ -64,6 +64,8 @@ def _ai_annotate_rectangles(
                 "shapes": shapes,
                 "imagePath": os.path.basename(image_file),
                 "imageData": None,
+                "imageHeight": image.shape[0],
+                "imageWidth": image.shape[1],
             },
             f,
             indent=2,