Merge pull request #1268 from roboflow/fix/scale-sv-detections-data-polygon

Fix/scale sv detections data polygon

Author: grzegorz-roboflow

inference/core/version.py

@@ -1,4 +1,4 @@
-__version__ = "0.49.1"
+__version__ = "0.49.2"
 
 
 if __name__ == "__main__":

inference/core/workflows/core_steps/common/utils.py

@@ -31,6 +31,7 @@
     PARENT_COORDINATES_KEY,
     PARENT_DIMENSIONS_KEY,
     PARENT_ID_KEY,
+    POLYGON_KEY_IN_SV_DETECTIONS,
     PREDICTION_TYPE_KEY,
     ROOT_PARENT_COORDINATES_KEY,
     ROOT_PARENT_DIMENSIONS_KEY,
@@ -389,6 +390,12 @@ def scale_sv_detections(
             scaled_detection_mask = np.sum(polygon_masks, axis=0) > 0
             scaled_masks.append(scaled_detection_mask)
         detections_copy.mask = np.array(scaled_masks)
+    if POLYGON_KEY_IN_SV_DETECTIONS in detections_copy.data:
+        detections_copy.data[POLYGON_KEY_IN_SV_DETECTIONS] = (
+            (detections_copy.data[POLYGON_KEY_IN_SV_DETECTIONS] * scale)
+            .round()
+            .astype(np.int32)
+        )
     if SCALING_RELATIVE_TO_PARENT_KEY in detections_copy.data:
         detections_copy[SCALING_RELATIVE_TO_PARENT_KEY] = (
             detections_copy[SCALING_RELATIVE_TO_PARENT_KEY] * scale

requirements/requirements.cli.txt

@@ -1,5 +1,6 @@
 requests>=2.32.0,<3.0.0
 docker>=7.0.0,<8.0.0
+click<8.2.0  # temporary pin until fastapi/typer#1145 is released
 typer>=0.9.0,<=0.12.5
 rich>=13.0.0,<13.10.0
 PyYAML~=6.0.0