fix ciou implementation bug (#20784)

keras-team · Jan 20, 2025 · 70125d4 · 70125d4
1 parent 35f76b8
commit 70125d4
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/keras/src/layers/preprocessing/image_preprocessing/bounding_boxes/iou.py b/keras/src/layers/preprocessing/image_preprocessing/bounding_boxes/iou.py
@@ -267,9 +267,9 @@ def compute_ciou(boxes1, boxes2, bounding_box_format, image_shape=None):
     )
 
     v = ops.squeeze(
-        ops.power(
-            (4 / math.pi**2)
-            * (ops.arctan(width_2 / height_2) - ops.arctan(width_1 / height_1)),
+        (4 / math.pi**2)
+        * ops.power(
+            (ops.arctan(width_2 / height_2) - ops.arctan(width_1 / height_1)),
             2,
         ),
         axis=-1,