o e @sZddlZddlZddlmZddlZddlZdejdejdejfddZGdd d Z dS) N)OptionalimagexyxyreturncCs4t|t}|\}}}}|||||f}|S)a Crops the given image based on the given bounding box. Args: image (np.ndarray): The image to be cropped, represented as a numpy array. xyxy (np.ndarray): A numpy array containing the bounding box coordinates in the format (x1, y1, x2, y2). Returns: (np.ndarray): The cropped image as a numpy array. Examples: ```python >>> import supervision as sv >>> detection = sv.Detections(...) >>> with sv.ImageSink(target_dir_path='target/directory/path') as sink: ... for xyxy in detection.xyxy: ... cropped_image = sv.crop_image(image=image, xyxy=xyxy) ... sink.save_image(image=image) ``` )nproundastypeint)rrx1y1x2y2 cropped_imgrF/var/www/myenv/lib/python3.10/site-packages/supervision/utils/image.py crop_image s rc@sReZdZ  ddededefddZdd Zdd ejd e efd dZ ddZ d S) ImageSinkFimage_{:05d}.pngtarget_dir_path overwriteimage_name_patterncCs||_||_||_d|_dS)a: Initialize a context manager for saving images. Args: target_dir_path (str): The target directory where images will be saved. overwrite (bool, optional): Whether to overwrite the existing directory. Defaults to False. image_name_pattern (str, optional): The image file name pattern. Defaults to "image_{:05d}.png". Examples: ```python >>> import supervision as sv >>> with sv.ImageSink(target_dir_path='target/directory/path', ... overwrite=True) as sink: ... for image in sv.get_video_frames_generator( ... source_path='source_video.mp4', stride=2): ... sink.save_image(image=image) ``` rN)rrr image_count)selfrrrrrr__init__(s zImageSink.__init__cCs@tj|jr|jrt|jt|j|St|j|SN)ospathexistsrrshutilrmtreemakedirs)rrrr __enter__Is   zImageSink.__enter__Nr image_namecCsD|dur |j|j}tj|j|}t|||jd7_dS)a: Save a given image in the target directory. Args: image (np.ndarray): The image to be saved. image_name (str, optional): The name to use for the saved image. If not provided, a name will be generated using the `image_name_pattern`. N) rformatrrrjoinrcv2imwrite)rrr" image_pathrrr save_imageSs   zImageSink.save_imagecCsdSrr)rexc_type exc_value exc_tracebackrrr__exit__dszImageSink.__exit__)Frr) __name__ __module__ __qualname__strboolrr!rndarrayrr)r-rrrrr's ! r) rrtypingrr&numpyrr3rrrrrrs