o eJ @sddlmZmZmZddlZddlmZddlZ dde j dee e fdee ddfd d Z   dd ee j d ee e fd eee dee e fdee ddf ddZdS))ListOptionalTupleN rgrayimagesizecmapreturncCsPtj|d|jdkrtj||dn tt|tjtdtdS)a Plots image using matplotlib. Args: image (np.ndarray): The frame to be displayed. size (Tuple[int, int]): The size of the plot. cmap (str): the colormap to use for single channel images. Examples: ```python >>> import cv2 >>> import supervision as sv >>> image = cv2.imread("path/to/image.jpg") %matplotlib inline >>> sv.plot_image(image=image, size=(16, 16)) ``` )figsizer offN) pltfigurendimimshowcv2cvtColor COLOR_BGR2RGBaxisshow)rr r rI/var/www/myenv/lib/python3.10/site-packages/supervision/utils/notebook.py plot_images    rimages grid_sizetitlesc Cs|\}}t|||krtdtj|||d\}}t|jD]=\} } | t|krW|| jdkr:| j|| |dn | t || tj |durW| t|krW| || | dqt dS)a: Plots images in a grid using matplotlib. Args: images (List[np.ndarray]): A list of images as numpy arrays. grid_size (Tuple[int, int]): A tuple specifying the number of rows and columns for the grid. titles (Optional[List[str]]): A list of titles for each image. Defaults to None. size (Tuple[int, int]): A tuple specifying the width and height of the entire plot in inches. cmap (str): the colormap to use for single channel images. Raises: ValueError: If the number of images exceeds the grid size. Examples: ```python >>> import cv2 >>> import supervision as sv >>> image1 = cv2.imread("path/to/image1.jpg") >>> image2 = cv2.imread("path/to/image2.jpg") >>> image3 = cv2.imread("path/to/image3.jpg") >>> images = [image1, image2, image3] >>> titles = ["Image 1", "Image 2", "Image 3"] %matplotlib inline >>> plot_images_grid(images, grid_size=(2, 2), titles=titles, size=(16, 16)) ``` ziThe number of images exceeds the grid size. Please increase the grid size or reduce the number of images.)nrowsncolsr r rNr)len ValueErrorrsubplots enumerateflatrrrrr set_titlerr) rrrr r rr figaxesidxaxrrrplot_images_grid)s'   r+)rr)Nrr)typingrrrrmatplotlib.pyplotpyplotrnumpynpndarrayintstrrr+rrrrs<   $