Mirrors/modelscope
1
0
Fork 0
mirror of https://github.com/modelscope/modelscope.git synced 2026-02-24 20:19:51 +01:00
Code Issues Packages Projects Releases Wiki Activity

fix bug to sync with master

Browse Source
This commit is contained in:
江袤
2023-03-01 14:22:28 +08:00
parent 6924c1583a
commit 8a8501b62b
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
modelscope/pipelines/cv/image_driving_perception_pipeline.py
View File

@@ -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