From 6093cbf7e12c8826fbd092a001b50b1645a2e9eb Mon Sep 17 00:00:00 2001 From: Danil Date: Fri, 17 Nov 2023 17:51:57 +0500 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. --- 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