mindspore.dataset.utils.imshow_det_bbox
- mindspore.dataset.utils.imshow_det_bbox(image, bboxes, labels, segm=None, class_names=None, score_threshold=0, bbox_color=(0, 255, 0), text_color=(203, 192, 255), mask_color=(128, 0, 128), thickness=2, font_size=0.8, show=True, win_name='win', wait_time=2000, out_file=None)[source]
Draw an image with given bboxes and class labels (with scores).
- Parameters
image (ndarray) – The image to be displayed, shaped (C, H, W) or (H, W, C), formatted RGB.
bboxes (ndarray) – Bounding boxes (with scores), shaped (N, 4) or (N, 5), data should be ordered with (N, x, y, w, h).
labels (ndarray) – Labels of bboxes, shaped (N, 1).
segm (ndarray) – The segmentation masks of image in M classes, shaped (M, H, W) (Default=None).
class_names (list[str], tuple[str], dict) – Names of each class to map label to class name (Default=None, only display label).
score_threshold (float) – Minimum score of bboxes to be shown (Default=0).
bbox_color (tuple(int)) – Color of bbox lines. The tuple of color should be in BGR order (Default=(0, 255 ,0), means ‘green’).
text_color (tuple(int)) – Color of texts. The tuple of color should be in BGR order (Default=(203, 192, 255), means ‘pink’).
mask_color (tuple(int)) – Color of mask. The tuple of color should be in BGR order (Default=(128, 0, 128), means ‘purple’).
thickness (int) – Thickness of lines (Default=2).
show (bool) – Whether to show the image (Default=True).
win_name (str) – The window name (Default=”win”).
wait_time (int) – Value of waitKey param (Default=2000, means display interval is 2000ms).
out_file (str, optional) – The filename to write the imagee (Default=None). File extension name is required to indicate the image compression type, e.g. ‘jpg’, ‘png’.
- Returns
The image with bboxes drawn on it.
- Return type
ndarray