From edc7c9e3d0c9021dcda050fe362934c3c728f9db Mon Sep 17 00:00:00 2001
From: PingqiLi <lipingqi2018@gmail.com>
Date: Wed, 22 May 2024 15:25:13 +0800
Subject: [PATCH] Modify the ssd/data.py to make it compatible with the
 mindspore v2.0+ version.

---
 examples/det/ssd/data.py | 10 +++++++++-
 1 file changed, 9 insertions(+), 1 deletion(-)

diff --git a/examples/det/ssd/data.py b/examples/det/ssd/data.py
index 31d82ca08..1aa06d220 100644
--- a/examples/det/ssd/data.py
+++ b/examples/det/ssd/data.py
@@ -1,3 +1,4 @@
+import inspect
 import os
 
 import cv2
@@ -175,14 +176,21 @@ def compose_map_func(img_id, image, annotation):
             output_columns = ["img_id", "image", "image_shape"]
             trans = [normalize_op, change_swap_op]
 
+        # Note: mindspore-2.0 delete the parameter column_order
+        sig = inspect.signature(ds.map)
+        pass_column_order = False if "kwargs" in sig.parameters else True
+
         ds = ds.map(
             operations=compose_map_func,
             input_columns=["img_id", "image", "annotation"],
             output_columns=output_columns,
-            column_order=output_columns,
+            column_order=output_columns if pass_column_order else None,
             python_multiprocessing=python_multiprocessing,
             num_parallel_workers=num_parallel_workers,
         )
+        if not pass_column_order:
+            ds.project(output_columns=output_columns)
+
         ds = ds.map(
             operations=trans,
             input_columns=["image"],