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