fix benchmark det_r50_vd_pse_v2_0 train error

PaddlePaddle · Nov 16, 2024 · d9a43c5 · d9a43c5
1 parent 7bbda2b
commit d9a43c5
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/ppocr/data/imaug/iaa_augment.py b/ppocr/data/imaug/iaa_augment.py
@@ -208,5 +208,5 @@ def __call__(self, data):
                 new_poly = transformed_keypoints[idx : idx + length]
                 new_polys.append(np.array([kp[:2] for kp in new_poly]))
                 idx += length
-            data["polys"] = new_polys
+            data["polys"] = np.array(new_polys)
         return data
diff --git a/tests/test_iaa_augment.py b/tests/test_iaa_augment.py
@@ -50,7 +50,9 @@ def test_iaa_augment_default(sample_image, sample_polys):
     assert isinstance(
         transformed_data["image"], np.ndarray
     ), "Image should be a numpy array"
-    assert isinstance(transformed_data["polys"], list), "Polys should be a list"
+    assert isinstance(
+        transformed_data["polys"], np.ndarray
+    ), "Polys should be a numpy array"
     assert transformed_data["image"].ndim == 3, "Image should be 3-dimensional"
 
     # Verify that the polygons have been transformed