Skip to content

Commit 1413201

Browse files
committed
修复了Nanodet中CKFT和SORT追踪模型无法使用的错误
1 parent 4c92db5 commit 1413201

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

motrackers/detectors/nanodet.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -75,4 +75,6 @@ def visualize(self, dets, meta, class_names, score_thres, wait=0):
7575
cv2.rectangle(image, (bb[0], y_label - label_height), (bb[0] + label_width, y_label + baseLine),
7676
(255, 255, 255), cv2.FILLED)
7777
cv2.putText(image, label, (bb[0], y_label), cv2.FONT_HERSHEY_SIMPLEX, 0.5, clr, 2)
78+
bboxes = np.array(bboxes).astype('int')
79+
confidences = np.array(confidences)
7880
return bboxes , confidences , class_ids , image

0 commit comments

Comments
 (0)