mmcv.visualization.imshow_det_bboxes¶

mmcv.visualization.imshow_det_bboxes(img: Union[str, numpy.ndarray], bboxes: numpy.ndarray, labels: numpy.ndarray, class_names: Optional[List[str]] = None, score_thr: float = 0, bbox_color: Union[mmcv.visualization.color.Color, str, tuple, int, numpy.ndarray] = 'green', text_color: Union[mmcv.visualization.color.Color, str, tuple, int, numpy.ndarray] = 'green', thickness: int = 1, font_scale: float = 0.5, show: bool = True, win_name: str = '', wait_time: int = 0, out_file: Optional[str] = None)[源代码]¶

Draw bboxes and class labels (with scores) on an image.

参数

img (str or ndarray) – The image to be displayed.
bboxes (ndarray) – Bounding boxes (with scores), shaped (n, 4) or (n, 5).
labels (ndarray) – Labels of bboxes.
class_names (list[str]) – Names of each classes.
score_thr (float) – Minimum score of bboxes to be shown.
bbox_color (Color or str or tuple or int or ndarray) – Color of bbox lines.
text_color (Color or str or tuple or int or ndarray) – Color of texts.
thickness (int) – Thickness of lines.
font_scale (float) – Font scales of texts.
show (bool) – Whether to show the image.
win_name (str) – The window name.
wait_time (int) – Value of waitKey param.
out_file (str or None) – The filename to write the image.

返回

The image with bboxes drawn on it.

返回类型

ndarray