^j0ddlZddlmZddlmZmZmZddlZddlZ ddl mZ ddl m Z mZddlmZddlmZededef Zd ed efd Ze ed dd ed efdZded efdZd ed efdZe ed dd ed efdZe ed dded efdZdeed ee j6e j8fdZdej d e j6e j8fdZde j6e j8d ej fdZy)N)Callable)AnyTypeVarcast) deprecatedvoid)Image) ImageTypeF.)bound annotate_funcreturnc tjdtdtdtdtdtf fd }t t |S)z Decorates `BaseAnnotator.annotate` implementations, converts scene to an image type used internally by the annotators, converts back when annotation is complete. Assumes the annotators modify the scene in-place. selfsceneargskwargsrct|tjr ||g|i|St|tjr4t |}||g|i|}|j t ||Stdt|NzUnsupported image type: ) isinstancenpndarrayr pillow_to_cv2paste cv2_to_pillow ValueErrortype)rrrrscene_np annotated_npr s g/var/www/ramen.bs-engineer-server.com/venv/lib/python3.12/site-packages/supervision/utils/conversion.pywrapperz2ensure_cv2_image_for_class_method..wrappers eRZZ ( u>t>v> > eU[[ )$U+H(xI$I&IL KK l3 4L3DK=ABB functoolswrapsrr rr r r!s` r !ensure_cv2_image_for_class_methodr'sY__]# Cc C) CC C3 C9 C$ C 7 r"z0.27.0z0.31.0)target deprecated_in remove_inc4ttt|SNrr rr s r ensure_cv2_image_for_annotationr/+ 4 & ''r"image_processing_func tjdtdtdtdtffd }t t |S)z Decorates image processing functions that accept np.ndarray, converting `image` to np.ndarray, converts back when processing is complete. Assumes the annotators do NOT modify the scene in-place. imagerrrct|tjr |g|i|St|tjr"t |}|g|i|}t |St dt|r)rrrr rrrr)r3rrr annotatedr1s r r!z9ensure_cv2_image_for_standalone_function..wrapper@st eRZZ ('??? ? eU[[ )!%(E,UDTDVDI + +3DK=ABBr")r$r%r rrr )r1r!s` r (ensure_cv2_image_for_standalone_functionr66sR__)* Cy C C C C+ C 7 r"c tjdtdtdtdtdtf fd }t t |S)z Decorates image processing functions that accept np.ndarray, converting `image` to PIL image, converts back when processing is complete. Assumes the annotators modify the scene in-place. rrrrrcBt|tjr9t|}||g|i|}tj|t ||St|t j rtt||g|i|Stdt|r) rrrrcopytorr rr rr)rrrr scene_pil annotated_pilr s r r!z2ensure_pil_image_for_class_method..wrapperYs eRZZ (%e,I)$ KDKFKM IIe]=9 :L eU[[ ) =u#Nt#Nv#NO O3DK=ABBr"r#r&s` r !ensure_pil_image_for_class_methodr<OsY__]# Cc C) CC C3 C9 C$ C 7 r"c4ttt|Sr,r-r.s r ensure_pil_image_for_annotationr>ir0r"c4ttt|Sr,r-)r1s r ensure_cv2_image_for_processingr@ts 4,- ..r"imagescg}|D]A}tt|tjr t|}|j |C|S)z Converts images provided either as Pillow images or OpenCV images into OpenCV format. Args: images: Images to be converted Returns: List of input images in OpenCV format (with order preserved). ) issubclassrr rappend)rAresultr3s r images_to_cv2rFsGF d5k5;; /!%(E e Mr"r3ctj|}tj|tj}|j tj dS)z Converts Pillow image into OpenCV image, handling RGB -> BGR conversion. Args: image: Pillow image (in RGB format). Returns: Input image converted to OpenCV format. F)copy)rarraycv2cvtColor COLOR_RGB2BGRastypeuint8)r3rs r rrs@ HHUOE LL 1 1 2E <<u< --r"cttj|tj}tj|S)z Converts OpenCV image into Pillow image, handling BGR -> RGB conversion. Args: image: OpenCV image (in BGR format). Returns: Input image converted to Pillow format. )rJrK COLOR_BGR2RGBr fromarray)r3s r rrs) LL 1 1 2E ??5 !!r") r$collections.abcrtypingrrrrJnumpyr numpy.typingnpt deprecaterrPILr supervision.draw.baser r r'r/r6r<r>r@listNDArrayrNrFrrr"r r]s~$%% &+ CxS)*6 , (((  ( 24 , (((  (  3 ///  / $y/d3;;rxx3H.I*..RXX)>.$ "RXX. "5;; "r"