Browse Source

cv/cvdet_master_coord_round and scores .6f

Link: https://code.alibaba-inc.com/Ali-MaaS/MaaS-lib/codereview/9631070
master
wendi.hwd yingda.chen 3 years ago
parent
commit
af95a9107a
1 changed files with 2 additions and 2 deletions
  1. +2
    -2
      modelscope/models/cv/object_detection/mmdet_model.py

+ 2
- 2
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

Loading…
Cancel
Save