Shortcuts

mmcv.visualization.imshow_det_bboxes

mmcv.visualization.imshow_det_bboxes(img: str | ndarray, bboxes: ndarray, labels: ndarray, class_names: List[str] | None = None, score_thr: float = 0, bbox_color: Color | str | tuple | int | ndarray = 'green', text_color: Color | str | tuple | int | ndarray = 'green', thickness: int = 1, font_scale: float = 0.5, show: bool = True, win_name: str = '', wait_time: int = 0, out_file: str | None = None)[source]

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

Parameters:
  • 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.

Returns:

The image with bboxes drawn on it.

Return type:

ndarray