From 8a8501b62bd1a18483c09bccb239d463da5a7690 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E6=B1=9F=E8=A2=A4?= Date: Wed, 1 Mar 2023 14:22:28 +0800 Subject: [PATCH] fix bug to sync with master --- modelscope/pipelines/cv/image_driving_perception_pipeline.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/modelscope/pipelines/cv/image_driving_perception_pipeline.py b/modelscope/pipelines/cv/image_driving_perception_pipeline.py index 0599277b..099d990c 100644 --- a/modelscope/pipelines/cv/image_driving_perception_pipeline.py +++ b/modelscope/pipelines/cv/image_driving_perception_pipeline.py @@ -98,7 +98,7 @@ class ImageDrivingPerceptionPipeline(Pipeline): det[:, :4] = scale_coords(inputs['img_hw'], det[:, :4], (h, w)).round() - results_dict[OutputKeys.BOXES].append(det[:, :4].cpu().numpy()) + results_dict[OutputKeys.BOXES] = (det[:, :4].cpu().numpy()) results_dict[OutputKeys.MASKS].append(da_seg_mask) results_dict[OutputKeys.MASKS].append(ll_seg_mask) return results_dict