From af95a9107a9558e2aef3ce0411bf2ac98329028d Mon Sep 17 00:00:00 2001 From: "wendi.hwd" Date: Wed, 3 Aug 2022 20:32:38 +0800 Subject: [PATCH] cv/cvdet_master_coord_round and scores .6f Link: https://code.alibaba-inc.com/Ali-MaaS/MaaS-lib/codereview/9631070 --- modelscope/models/cv/object_detection/mmdet_model.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/modelscope/models/cv/object_detection/mmdet_model.py b/modelscope/models/cv/object_detection/mmdet_model.py index 53745f75..51f05e47 100644 --- a/modelscope/models/cv/object_detection/mmdet_model.py +++ b/modelscope/models/cv/object_detection/mmdet_model.py @@ -87,7 +87,7 @@ class DetectionModel(TorchModel): return None, None, None bboxes = bbox_result[inds, :] labels = labels[inds] - scores = bboxes[:, 4] - bboxes = bboxes[:, 0:4] + scores = np.around(bboxes[:, 4], 6) + bboxes = (bboxes[:, 0:4]).astype(int) labels = [self.class_names[i_label] for i_label in labels] return bboxes, scores, labels