Mirrors/modelscope
1
0
Fork 0
mirror of https://github.com/modelscope/modelscope.git synced 2026-09-01 19:49:03 +02:00
Code Issues Packages Projects Releases Wiki Activity

cv/cvdet_master_coord_round and scores .6f

Browse Source 
Link: https://code.alibaba-inc.com/Ali-MaaS/MaaS-lib/codereview/9631070
This commit is contained in:
wendi.hwd
2022-08-03 20:32:38 +08:00
committed by yingda.chen
parent 064f1041a9
commit af95a9107a
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
modelscope/models/cv/object_detection/mmdet_model.py
View File

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