From 6ff3c936135409e33d73f8dd31161a55824e9610 Mon Sep 17 00:00:00 2001 From: Danil Date: Thu, 14 Dec 2023 06:04:43 +0300 Subject: [PATCH] There was an error when using a model from the mmyolo repository during image preprocessing, provided that an array was supplied as input. If you input the path to the image, everything works fine. (#2554) --- mmdeploy/codebase/mmdet/deploy/object_detection.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mmdeploy/codebase/mmdet/deploy/object_detection.py b/mmdeploy/codebase/mmdet/deploy/object_detection.py index 889af233eb..3af3d29015 100644 --- a/mmdeploy/codebase/mmdet/deploy/object_detection.py +++ b/mmdeploy/codebase/mmdet/deploy/object_detection.py @@ -60,7 +60,7 @@ def process_model_config(model_cfg: Config, if isinstance(imgs[0], np.ndarray): cfg = cfg.copy() # set loading pipeline type - cfg.test_pipeline[0].type = 'LoadImageFromNDArray' + cfg.test_pipeline[0].type = 'mmdet.LoadImageFromNDArray' pipeline = cfg.test_pipeline