Skip to content

GitLab

Commit 3e1befe1 authored by Kai Chen's avatar Kai Chen
Browse files

bug fix

parent 97f9febd
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
mmcv/visualization/image.py
View file @ 3e1befe1
...@@ -107,7 +107,7 @@ def imshow_det_bboxes(img, ...@@ -107,7 +107,7 @@ def imshow_det_bboxes(img,
scores = bboxes[:, -1] scores = bboxes[:, -1]
inds = scores > score_thr inds = scores > score_thr
bboxes = bboxes[inds, :] bboxes = bboxes[inds, :]
labels = labels[inds, :] labels = labels[inds]
bbox_color = color_val(bbox_color) bbox_color = color_val(bbox_color)
text_color = color_val(text_color) text_color = color_val(text_color)
...@@ -120,7 +120,7 @@ def imshow_det_bboxes(img, ...@@ -120,7 +120,7 @@ def imshow_det_bboxes(img,
img, left_top, right_bottom, bbox_color, thickness=thickness) img, left_top, right_bottom, bbox_color, thickness=thickness)
label_text = class_names[ label_text = class_names[
label] if class_names is not None else 'cls {}'.format(label) label] if class_names is not None else 'cls {}'.format(label)
if bbox.shape[1] > 4: if len(bbox) > 4:
label_text += '|{:.02f}'.format(bbox[-1]) label_text += '|{:.02f}'.format(bbox[-1])
cv2.putText(img, label_text, (bbox_int[0], bbox_int[1] - 2), cv2.putText(img, label_text, (bbox_int[0], bbox_int[1] - 2),
cv2.FONT_HERSHEY_COMPLEX, font_scale, text_color) cv2.FONT_HERSHEY_COMPLEX, font_scale, text_color)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment