From 1a127ed997d98b1024e5bcf7ee20a46868bea996 Mon Sep 17 00:00:00 2001 From: "leyuan.hjy" Date: Thu, 17 Nov 2022 18:11:40 +0800 Subject: [PATCH] fix outputs none MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit fix 输出为空的情况 Link: https://code.alibaba-inc.com/Ali-MaaS/MaaS-lib/codereview/10725840 --- .../realtime_object_detection/realtime_video_detector.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/modelscope/models/cv/realtime_object_detection/realtime_video_detector.py b/modelscope/models/cv/realtime_object_detection/realtime_video_detector.py index 3830fb42..ebe0221b 100644 --- a/modelscope/models/cv/realtime_object_detection/realtime_video_detector.py +++ b/modelscope/models/cv/realtime_object_detection/realtime_video_detector.py @@ -7,6 +7,7 @@ import time import cv2 import json +import numpy as np import torch from tqdm import tqdm @@ -87,13 +88,17 @@ class RealtimeVideoDetector(TorchModel): self.nmsthre, class_agnostic=True) - if len(outputs) == 1: + if len(outputs) == 1 and (outputs[0] is not None): bboxes = outputs[0][:, 0:4].cpu().numpy() / self.ratio scores = outputs[0][:, 5].cpu().numpy() labels = outputs[0][:, 6].cpu().int().numpy() pred_label_names = [] for lab in labels: pred_label_names.append(self.label_mapping[lab]) + else: + bboxes = np.asarray([]) + scores = np.asarray([]) + pred_label_names = np.asarray([]) return bboxes, scores, pred_label_names