support uniform_output_enabled

paddlex/repo_apis/PaddleClas_api/cls/model.py

@@ -114,6 +114,7 @@ class ClsModel(BaseModel):
                 config._update_amp(amp)
 
             # PDX related settings
+            config.update(["Global.uniform_output_enabled=True"])
             config.update([f"Global.pdx_model_name={self.name}"])
             hpi_config_path = self.model_info.get("hpi_config_path", None)
             config.update([f"Global.hpi_config_path={hpi_config_path}"])

paddlex/repo_apis/PaddleDetection_api/instance_seg/model.py

@@ -125,6 +125,7 @@ class InstanceSegModel(BaseModel):
                 cli_args.append(CLIArgument("--enable_ce", enable_ce))
 
         # PDX related settings
+        config.update({"uniform_output_enabled": True})
         config.update({"pdx_model_name": self.name})
         hpi_config_path = self.model_info.get("hpi_config_path", None)
         if hpi_config_path:

paddlex/repo_apis/PaddleDetection_api/object_det/model.py

@@ -126,6 +126,7 @@ class DetModel(BaseModel):
                 cli_args.append(CLIArgument("--enable_ce", enable_ce))
 
         # PDX related settings
+        config.update({"uniform_output_enabled": True})
         config.update({"pdx_model_name": self.name})
         hpi_config_path = self.model_info.get("hpi_config_path", None)
         if hpi_config_path:

paddlex/repo_apis/PaddleOCR_api/text_rec/model.py

@@ -132,6 +132,7 @@ class TextRecModel(BaseModel):
                     os.environ[env_name] = str(env_value)
 
         # PDX related settings
+        config.update({"Global.uniform_output_enabled": True})
         config.update({"Global.pdx_model_name": self.name})
         hpi_config_path = self.model_info.get("hpi_config_path", None)
         config.update({"Global.hpi_config_path": hpi_config_path})

paddlex/repo_apis/PaddleSeg_api/seg/model.py

@@ -165,6 +165,7 @@ class SegModel(BaseModel):
                 cli_args.append(CLIArgument("--seed", seed))
 
         # PDX related settings
+        config.set_val("uniform_output_enabled", True)
         config.set_val("pdx_model_name", self.name)
         hpi_config_path = self.model_info.get("hpi_config_path", None)
         if hpi_config_path: