^j ddlmZddlZddlmZddlmZddlm Z ddl m Z  d d dZ d d dZ y) ) annotationsN)Image) ImageType) pillow_to_cv2ct|tjr t|}tj||j dk(rtj ||n7tj tj|tjtjdtjy)a Plots image using matplotlib. Args: image: The frame to be displayed ImageType is a flexible type, accepting either `numpy.ndarray` or `PIL.Image.Image`. size: The size of the plot in inches. cmap: the colormap to use for single channel images. Examples: ```pycon >>> import cv2 >>> import numpy as np >>> import matplotlib >>> matplotlib.use('Agg') # Prevents the GUI window from popping up >>> import supervision as sv >>> image = np.zeros((100, 100, 3), dtype=np.uint8) >>> sv.plot_image(image=image, size=(16, 16)) ... ``` )figsizecmapoffN) isinstancerrpltfigurendimimshowcv2cvtColor COLOR_BGR2RGBaxisshow)imagesizer s e/var/www/ramen.bs-engineer-server.com/venv/lib/python3.12/site-packages/supervision/utils/notebook.py plot_imager sr2%%e$JJt zzQ 5t$ 3<<s'8'89:HHUOHHJc|\}}t|D].\}}t|tjs!t|||<0t |||zkDr t dt j|||\} } t| jD]\}} |t |kr||jdk(r| j|||n6| jtj||tj|"|t |kr| j||| jdt j y)a Plots images in a grid using matplotlib. Args: images: A list of images as ImageType is a flexible type, accepting either `numpy.ndarray` or `PIL.Image.Image`. grid_size: A tuple specifying the number of rows and columns for the grid. titles: A list of titles for each image. Defaults to None. size: A tuple specifying the width and height of the entire plot in inches. cmap: the colormap to use for single channel images. Raises: ValueError: If the number of images exceeds the grid size. Examples: ```pycon >>> import cv2 >>> import numpy as np >>> import matplotlib >>> matplotlib.use('Agg') # Prevents the GUI window from popping up >>> import supervision as sv >>> from PIL import Image >>> image1 = np.zeros((100, 100, 3), dtype=np.uint8) >>> image2 = Image.new('RGB', (100, 100)) >>> image3 = np.zeros((100, 100, 3), dtype=np.uint8) >>> images = [image1, image2, image3] >>> titles = ["Image 1", "Image 2", "Image 3"] >>> sv.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.)nrowsncolsrr r Nr ) enumerater rrlen ValueErrorrsubplotsflatrrrrr set_titlerr) images grid_sizetitlesrr rridximg_figaxesaxs rplot_images_gridr-2s$TLE5f%-S c5;; '',F3K- 6{UU]" /  EEJD$TYY' R V c{1$ &+D 1 #,,vc{C4E4EFG!cCK&7 VC[)  HHJr) r/gray)rrrtuple[int, int]r str | NonereturnNone)Nr.r0) r%zlist[ImageType]r&r1r'zlist[str] | Nonerr1r r2r3r4) __future__rrmatplotlib.pyplotpyplotrPILrsupervision.draw.basersupervision.utils.conversionrrr-rrr<s" +6LR$ $+$>H$ $T $$ C CC C  C  C  Cr