fix outputs none

fix 输出为空的情况
        Link: https://code.alibaba-inc.com/Ali-MaaS/MaaS-lib/codereview/10725840
This commit is contained in:
leyuan.hjy
2022-11-17 18:11:40 +08:00
committed by yingda.chen
parent ed9d2b5436
commit 1a127ed997

View File

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