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

[to #42322933] 视频全景分割 修复一个测试bug

Browse Source 
Link: https://code.alibaba-inc.com/Ali-MaaS/MaaS-lib/codereview/11660389
This commit is contained in:
james.wjg
2023-02-14 10:28:01 +00:00
committed by yingda.chen
parent 5fe9d0682e
commit 295ec86d4e
1 changed files with 6 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
modelscope/models/cv/video_panoptic_segmentation/video_k_net.py
View File

@@ -331,6 +331,8 @@ class VideoKNet(TorchModel):
thing_masks_for_tracking_final, 0)
thing_masks_for_tracking = thing_masks_for_tracking_final
thing_masks_for_tracking_with_semantic_filter = thing_masks_for_tracking_final * semantic_thing
else:
things_bbox_for_tracking = []
if len(things_labels_for_tracking) == 0:
track_feats = None
@@ -368,10 +370,12 @@ class VideoKNet(TorchModel):
semantic_map, binary_masks, labels = self.get_semantic_seg(
panoptic_seg, segments_info)
vis_tracker = None
vis_sem = None
from .visualizer import trackmap2rgb, cityscapes_cat2rgb, draw_bbox_on_img
vis_tracker = trackmap2rgb(track_maps)
vis_sem = cityscapes_cat2rgb(semantic_map)
if len(things_labels_for_tracking):
vis_tracker = trackmap2rgb(track_maps)
vis_sem = cityscapes_cat2rgb(semantic_map)
vis_tracker = draw_bbox_on_img(
vis_tracker,
things_bbox_for_tracking.cpu().numpy())