^j=ddlmZddlZddlZddlmZddlmZddlZ ddl mZ ddl m Z mZddlmZddlmZddlmZmZdd lmZej0Zd ZGd d eZej8f d d Z d!dZd"dZej8f d#dZ d$d%dZ!d&dZ"e e"ddd&dZ# d'dZ$ d(dZ%GddZ&d)dZ'd*dZ(d+dZ)d,dZ*d,dZ+y)-) annotationsN)Enum)Any) deprecatedvoid)CLASS_NAME_DATA_FIELD) Detections)Color ColorPalette)Positionc.eZdZdZdZdZdZeddZy) ColorLookupa Enumeration class to define strategies for mapping colors to annotations. This enum supports three different lookup strategies: - `INDEX`: Colors are determined by the index of the detection within the scene. - `CLASS`: Colors are determined by the class label of the detected object. - `TRACK`: Colors are determined by the tracking identifier of the object. indexclasstrackc.ttd|S)Nc|jSN)value)cs g/var/www/ramen.bs-engineer-server.com/venv/lib/python3.12/site-packages/supervision/annotators/utils.pyz"ColorLookup.list..%s !'')listmap)clss rrzColorLookup.list#sC)3/00rN)return list[str]) __name__ __module__ __qualname____doc__INDEXCLASSTRACK classmethodrrrrrs* E E E11rrcd|t|k\rtd|dt|t|tjrHt|t|k7r#tdt|dt|t ||S|t jk(r|S|t jk(r/|j tdt |j|S|t jk(r/|j tdt |j|Std|)NzDetection index z+ is out of bounds for detections of length zLength of color lookup z% does not match length of detections zCould not resolve color by class because Detections do not have class_id. If using an annotator, try setting color_lookup to sv.ColorLookup.INDEX or sv.ColorLookup.TRACK.zCould not resolve color by track because Detections do not have tracker_id. Did you call tracker.update_with_detections(...) before annotating?z#Unsupported color lookup strategy: ) len ValueError isinstancenpndarrayintrr$r%class_idr& tracker_id detections detection_idx color_lookups rresolve_color_idxr6(sK J'}o.99>> import numpy as np >>> from supervision.annotators.utils import snap_boxes >>> xyxy = np.array([ ... [-10, 10, 30, 50], # Off left edge ... [310, 200, 350, 250], # Off right edge ... [100, -20, 150, 30], # Off top edge ... [200, 220, 250, 270], # Off bottom edge ... [-20, 10, 350, 50], # Wider than frame (370 vs 320) ... [10, -20, 30, 260] # Taller than frame (280 vs 240) ... ]) >>> resolution_wh = (320, 240) >>> snapped_boxes = snap_boxes(xyxy=xyxy, resolution_wh=resolution_wh) >>> snapped_boxes array([[ 0., 10., 40., 50.], [280., 190., 320., 240.], [100., 0., 150., 50.], [200., 190., 250., 240.], [ 0., 10., 370., 50.], [ 10., 0., 30., 280.]], dtype=float32) ``` Nrr8)r-copynewaxisastypefloat32) xyxy resolution_whresultrXheight left_overflowright_overflow right_shift top_overflowbottom_overflow bottom_shifts r snap_boxesr sRXWWT]F!ME61a4L1$M =!Aa%  F=!A#+=$>> $n1)=>N&!233K >1Qq5 ![BJJ%??!!Q$?OF?A#566L ?AaE !"l1bjj=&AA" == $$rcNeZdZddejf ddZddZddZy) TraceNrc||_||_||_tjgt |_tjdtj|_ tjgt |_ y)Ndtype)rr8) current_frame_idmax_sizeanchorr-arrayr/frame_idemptyrxyr1)selfrstart_frame_idrs r__init__zTrace.__init__MsT !/   3/ ((64((2S1rcJtjt||jt}tj |j |g|_tj |j|j|jg|_|j tdtj |j|jg|_ tj|j }|jd|jcxkrt|krjnng|j|jz dz}|j |k\}|j ||_|j||_|j||_ |xjdz c_y)NrzNCould not put detections into Trace because Detections do not have tracker_id.rrz)r-fullr*rr/ concatenaterrget_anchors_coordinatesrr1r+uniquer)rr3runique_frame_idmax_allowed_frame_idfiltering_masks rputz Trace.put[sF)+  OT22#*  x'@A ..224;;?    (5  ..$//:;P;P)QR))DMM2 == $T]])QS=Q)Q#'#8#84==#H1#L !]].BBN MM.9DMggn-DG"oon=DO "rc|j|j|k(jjtj d}|S)NF)r|)rr1r|r~r-r)rr1filtereds rgetz Trace.getysA GGDOOz1 2 7 7 9 @ @RW @ X r)r int | Nonerr/rr rNone)r3r rr)r1r/r%np.ndarray[Any, np.dtype[np.float32]])r r!r"r r<rrrr(rrrrLsF $#?? 2 2 2 2  2#<rrcd|jjd}t|dk(r|dz }t|dk7rtd| t |ddd}t |dd d}t |d dd}t |ddd}||||fS#t$r}td ||d }~wwxYw) z Converts a hex color string (e.g. "#FF00FF" or "#FF00FF80") to an RGBA tuple. Args: hex_color: A hex color string. Returns: RGBA values in range 0-255. Raises: ValueError: If the format is invalid. #FFzInvalid hex color format: rr8r{zInvalid hex digits in N)striplstripr*r+r/) hex_colorrgbaexcs r hex_to_rgbars!((-I 9~T  9~5i[ABBH  !A #  !A #  !A #  !A # q!Q< H1)=>CGHszrgba_to_hex..s =1a3 =sz1RGBA must be a 4-tuple with values between 0-255.z#{:02X}{:02X}{:02X}{:02X})r*allr+format)rgbas r rgba_to_hexrs< 4yA~S = ==LMM - & - -t 44rc\ttjd|jS)a# Checks if a given string is a valid hex color. Args: hex_color: A hex color string with an optional leading "#". Supports 6-digit (RGB) or 8-digit (RGBA) formats. Returns: True if the string is a valid 6- or 8-digit hex color, otherwise False. z!#?[0-9A-Fa-f]{6}([0-9A-Fa-f]{2})?)boolre fullmatchr)rs r is_valid_hexrs"  A9??CTU VVrc@tdt||z ||z dzS)a Computes a blur kernel size proportional to the shorter side of a bounding box. Args: x1: Left edge of the bounding box. y1: Top edge of the bounding box. x2: Right edge of the bounding box. y2: Bottom edge of the bounding box. Returns: Kernel size as one-third of the shorter dimension, minimum 1. Examples: ```pycon >>> calculate_dynamic_kernel_size(0, 0, 90, 60) 20 ``` rzrymaxminx1y1x2y2s rcalculate_dynamic_kernel_sizer%( q#b2grBw'1, --rc@tdt||z ||z dzS)a Computes a pixelation size proportional to the shorter side of a bounding box. Args: x1: Left edge of the bounding box. y1: Top edge of the bounding box. x2: Right edge of the bounding box. y2: Bottom edge of the bounding box. Returns: Pixel size as one-half of the shorter dimension, minimum 1. Examples: ```pycon >>> calculate_dynamic_pixel_size(0, 0, 90, 60) 30 ``` rzr8rrs rcalculate_dynamic_pixel_sizerrr)r3r r4r/r5"ColorLookup | npt.NDArray[np.int_]rr/)rCtuple[int, int]rDrrEr rtuple[int, int, int, int])rNColor | ColorPaletterOr/rr ) rNrr3r r4r/r5rrr r)rcrrdrrr)rllist[str] | Noner3r rr)r3r rvrrr)rrrrrr)rr\rr)rrrr\)rr\rr) rr/rr/rr/rr/rr/), __future__rrr`enumrtypingrnumpyr- numpy.typingnpt deprecaterrsupervision.configrsupervision.detection.corer supervision.draw.colorr r supervision.geometry.corer GREYrSrRrr%r6rJrPrTrirmrrrwrrrrrrrr(rrrs}" &416.jj1$1,8C7H7H%K%K%K5%K %KP2 '2 2 2  2 j8C7H7H 4 4445 4  4(+\ *    +;>?% /?%"?%+?%D11h85$ W...r