Mirrors/modelscope
1
0
Fork 0
mirror of https://github.com/modelscope/modelscope.git synced 2025-12-25 12:39:25 +01:00
Code Issues Packages Projects Releases Wiki Activity

[to #42322933] convert output boxes from tensor to numpy according to demo-service rules

Browse Source 
[to #42322933] convert output boxes from tensor to numpy according to demo-service rules

Link: https://code.alibaba-inc.com/Ali-MaaS/MaaS-lib/codereview/11654774
This commit is contained in:
maojialiang.mjl
2023-02-14 08:35:11 +00:00
committed by wenmeng.zwm
parent e7cb55d4a6
commit 2b8236e6d0
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

@@ -94,7 +94,7 @@ class ImageDrivingPerceptionPipeline(Pipeline):
# Rescale boxes from img_size to (720, 1280)
det[:, :4] = scale_coords(inputs['img_hw'], det[:, :4]).round()
results_dict[OutputKeys.BOXES].append(det[:, :4])
results_dict[OutputKeys.BOXES].append(det[:, :4].cpu().numpy())
results_dict[OutputKeys.MASKS].append(da_seg_mask)
results_dict[OutputKeys.MASKS].append(ll_seg_mask)
return results_dict