From bc1b6440cd4e9eb85d02a706bbe263aeec4e66f9 Mon Sep 17 00:00:00 2001
From: AllentDan <41138331+AllentDan@users.noreply.github.com>
Date: Wed, 8 Feb 2023 13:47:47 +0800
Subject: [PATCH] fix mmocr create input error when the arg is array (#1721)

---
 mmdeploy/codebase/mmocr/deploy/text_recognition.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/mmdeploy/codebase/mmocr/deploy/text_recognition.py b/mmdeploy/codebase/mmocr/deploy/text_recognition.py
index 475fc24bb..cf943560c 100644
--- a/mmdeploy/codebase/mmocr/deploy/text_recognition.py
+++ b/mmdeploy/codebase/mmocr/deploy/text_recognition.py
@@ -29,6 +29,9 @@ def process_model_config(model_cfg: mmengine.Config,
     Returns:
         mmengine.Config: the model config after processing.
     """
+    if isinstance(imgs[0], np.ndarray):
+        # set loading pipeline type
+        model_cfg.test_pipeline[0].type = 'LoadImageFromNDArray'
     test_pipeline = model_cfg._cfg_dict.test_pipeline
     for i, transform in enumerate(test_pipeline):
         if transform.type == 'PackTextRecogInputs':