From 295ec86d4e7281194358d0a09d7329761cb2b984 Mon Sep 17 00:00:00 2001 From: "james.wjg" Date: Tue, 14 Feb 2023 10:28:01 +0000 Subject: [PATCH] =?UTF-8?q?[to=20#42322933]=20=E8=A7=86=E9=A2=91=E5=85=A8?= =?UTF-8?q?=E6=99=AF=E5=88=86=E5=89=B2=20=E4=BF=AE=E5=A4=8D=E4=B8=80?= =?UTF-8?q?=E4=B8=AA=E6=B5=8B=E8=AF=95bug=20=20=20=20=20=20=20=20=20Link:?= =?UTF-8?q?=20https://code.alibaba-inc.com/Ali-MaaS/MaaS-lib/codereview/11?= =?UTF-8?q?660389?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../models/cv/video_panoptic_segmentation/video_k_net.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/modelscope/models/cv/video_panoptic_segmentation/video_k_net.py b/modelscope/models/cv/video_panoptic_segmentation/video_k_net.py index d3b15657..8f92d4a9 100644 --- a/modelscope/models/cv/video_panoptic_segmentation/video_k_net.py +++ b/modelscope/models/cv/video_panoptic_segmentation/video_k_net.py @@ -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())