FPh@>ddlmZddlmZddlmZddlmZddlZ ddl m Z m Z m Z mZmZmZmZdZed Zed Zgd Zd ZGd dZGddZy))abc)repeat)Number)ListN) ltwh2xywh ltwh2xyxyresample_segments xywh2ltwh xywh2xyxy xyxy2ltwh xyxy2xywhcfd}|S)zFrom PyTorch internals.cft|tjr|Stt |S)z2Parse bounding boxes format between XYWH and LTWH.) isinstancerIterabletupler)xns eC:\Users\daisl\Desktop\realtime-object-detection\venv\Lib\site-packages\ultralytics/utils/instance.pyparsez_ntuple..parses&q#,,/qHU6!Q<5HH)rrs` r_ntupler sI Lr)xyxyxywhltwh)BboxescbeZdZdZd ddZdZdZdZdZd Z e dd e dddfd Z dd Z y)r a A class for handling bounding boxes. The class supports various bounding box formats like 'xyxy', 'xywh', and 'ltwh'. Bounding box data should be provided in numpy arrays. Attributes: bboxes (numpy.ndarray): The bounding boxes stored in a 2D numpy array. format (str): The format of the bounding boxes ('xyxy', 'xywh', or 'ltwh'). Note: This class does not handle normalization or denormalization of bounding boxes. returnNc|tvsJd|dt|jdk(r |dddfn|}|jdk(sJ|jddk(sJ||_||_y)zJInitializes the Bboxes class with bounding box data in a specified format.Invalid bounding box format: , format must be one of rNrr)_formatsndimshapebboxesformat)selfr)r*s r__init__zBboxes.__init__1sv!m%B6(Jbckbl#mm!$*KK1$4a&{{a||A!###  rc8|tvsJd|dt|j|k(ry|jdk(r|dk(rtnt}n2|jdk(r|dk(rtnt }n|dk(rt nt}||j|_||_y)z6Converts bounding box format from one type to another.r$r%Nrr) r&r*rr r r r rr))r+r*funcs rconvertzBboxes.convert;s!m%B6(Jbckbl#mm! ;;&  [[F " && 09iD [[F " && 09iD && 09iD4;;'  rc|jd|jdddf|jdddfz |jdddf|jdddfz zS)zReturn box areas.rNrrr)r/r)r+s rareasz Bboxes.areasIsZ V AqD!DKK1$55$++ad:KdkkZ[]^Z^N_:_``rczt|tr t|}t|ttfsJt |dk(sJ|j dddfxx|dzcc<|j dddfxx|dzcc<|j dddfxx|dzcc<|j dddfxx|dzcc<y)zZ Args: scale (tuple | list | int): the scale for four coords. rNrrrr1rr to_4tuplerlistlenr))r+scales rmulz Bboxes.mul^s eV $e$E%%///5zQ AqDU1X% AqDU1X% AqDU1X% AqDU1X%rczt|tr t|}t|ttfsJt |dk(sJ|j dddfxx|dz cc<|j dddfxx|dz cc<|j dddfxx|dz cc<|j dddfxx|dz cc<y)z\ Args: offset (tuple | list | int): the offset for four coords. rNrrrr1r5)r+offsets raddz Bboxes.addls ff %v&F&5$-0006{a AqDVAY& AqDVAY& AqDVAY& AqDVAY&rc,t|jS)zReturn the number of boxes.r8r)r2s r__len__zBboxes.__len__z4;;r boxes_listc.t|ttfsJ|s|tjdSt d|DsJt |dk(r|dS|tj|Dcgc]}|jc}|Scc}w)a Concatenate a list of Bboxes objects into a single Bboxes object. Args: boxes_list (List[Bboxes]): A list of Bboxes objects to concatenate. axis (int, optional): The axis along which to concatenate the bounding boxes. Defaults to 0. Returns: Bboxes: A new Bboxes object containing the concatenated bounding boxes. Note: The input should be a list or tuple of Bboxes objects. rc3<K|]}t|tywN)rr ).0boxs r z%Bboxes.concatenate..sAjs:c6*jraxis) rr7rnpemptyallr8 concatenater))clsrBrKbs rrOzBboxes.concatenate~s *tUm444rxx{# #AjAAAA z?a a= 2>>Z"@Z188Z"@tLMM"@s2B ct|tr(t|j|j ddS|j|}|j dk(s Jd|dt|S)a Retrieve a specific bounding box or a set of bounding boxes using indexing. Args: index (int, slice, or np.ndarray): The index, slice, or boolean array to select the desired bounding boxes. Returns: Bboxes: A new Bboxes object containing the selected bounding boxes. Raises: AssertionError: If the indexed bounding boxes do not form a 2-dimensional matrix. Note: When using boolean indexing, make sure to provide a boolean array with the same length as the number of bounding boxes. rrzIndexing on Bboxes with z failed to return a matrix!)rintr r)viewr')r+indexrQs r __getitem__zBboxes.__getitem__sg$ eS !$++e,11!R89 9 KK vv{Y6ug=XYY{ayr)rr"Nr)r"r )__name__ __module__ __qualname____doc__r,r/r3r:r=r@ classmethodrrOrWrrrr r "sU  a* & ' NT(^NNN0rr ceZdZdZdddZdZedZddZdZ d Z d Z dd Z d Z d ZdZdZddZdZeddedddfdZedZy) Instancesa Container for bounding boxes, segments, and keypoints of detected objects in an image. Attributes: _bboxes (Bboxes): Internal object for handling bounding box operations. keypoints (ndarray): keypoints(x, y, visible) with shape [N, 17, 3]. Default is None. normalized (bool): Flag indicating whether the bounding box coordinates are normalized. segments (ndarray): Segments array with shape [N, 1000, 2] after resampling. Args: bboxes (ndarray): An array of bounding boxes with shape [N, 4]. segments (list | ndarray, optional): A list or array of object segments. Default is None. keypoints (ndarray, optional): An array of keypoints with shape [N, 17, 3]. Default is None. bbox_format (str, optional): The format of bounding boxes ('xywh' or 'xyxy'). Default is 'xywh'. normalized (bool, optional): Whether the bounding box coordinates are normalized. Default is True. Examples: ```python # Create an Instances object instances = Instances( bboxes=np.array([[10, 10, 30, 30], [20, 20, 40, 40]]), segments=[np.array([[5, 5], [10, 10]]), np.array([[15, 15], [20, 20]])], keypoints=np.array([[[5, 5, 1], [10, 10, 1]], [[15, 15, 1], [20, 20, 1]]]) ) ``` Note: The bounding box format is either 'xywh' or 'xyxy', and is determined by the `bbox_format` argument. This class does not perform input validation, and it assumes the inputs are well-formed. Nr"c|g}t|||_||_||_t |dkDr*t |}t j|d}||_ yt jdt j}||_ y)z Args: bboxes (ndarray): bboxes with shape [N, 4]. segments (list | ndarray): segments. keypoints (ndarray): keypoints(x, y, visible) with shape [N, 17, 3]. N)r)r*rrJ)rir)dtype) r _bboxes keypoints normalizedr8r rLstackzerosfloat32segments)r+r)rird bbox_formatres rr,zInstances.__init__sw  HVK@ "$ x=1 (2Hxxq1H! xx BJJ?H  rc<|jj|y)zConvert bounding box format.r*N)rcr/)r+r*s r convert_bboxzInstances.convert_bboxs F+rc6|jjS)z%Calculate the area of bounding boxes.)rcr3r2s r bbox_areaszInstances.bbox_areass||!!##rc|jj||||f|ry|jdxx|zcc<|jdxx|zcc<|j/|jdxx|zcc<|jdxx|zcc<yy)zHThis might be similar with denormalize func but without normalized sign.r9N.r.r)rcr:rird)r+scale_wscale_h bbox_onlys rr9zInstances.scales '7CD   f( f( >> % NN6 "g - " NN6 "g - " &rc<|jsy|jj||||f|jdxx|zcc<|jdxx|zcc<|j.|jdxx|zcc<|jdxx|zcc<d|_y)zHDenormalizes boxes, segments, and keypoints from normalized coordinates.NrqrrrsFrercr:rirdr+whs r denormalizezInstances.denormalizes  1a|, f" f" >> % NN6 "a ' " NN6 "a ' "rcT|jry|jjd|z d|z d|z d|z f|jdxx|zcc<|jdxx|zcc<|j.|jdxx|zcc<|jdxx|zcc<d|_y)zFNormalize bounding boxes, segments, and keypoints to image dimensions.NrrqrrrsTrxrys r normalizezInstances.normalizes ??  Aq1ua!eQU;< f" f" >> % NN6 "a ' " NN6 "a ' "rc<|jrJd|jj||||f|jdxx|z cc<|jdxx|z cc<|j/|jdxx|z cc<|jdxx|z cc<yy)z!Handle rect and mosaic situation.z1you should add padding with absolute coordinates.)r<rrrsN)rercr=rird)r+padwpadhs r add_paddingzInstances.add_paddings??W$WW" tT4 89 f% f% >> % NN6 "d * " NN6 "d * " &rct|jr|j|n |j}|j|j|nd}|j|}|jj }t |||||jS)aB Retrieve a specific instance or a set of instances using indexing. Args: index (int, slice, or np.ndarray): The index, slice, or boolean array to select the desired instances. Returns: Instances: A new Instances object containing the selected bounding boxes, segments, and keypoints if present. Note: When using boolean indexing, make sure to provide a boolean array with the same length as the number of instances. N)r)rirdrjre)r8rirdr)rcr*r`re)r+rVrirdr)rjs rrWzInstances.__getitem__sy ,/t}}+=4=='4==-1^^-GDNN5)T U#ll)) #   rc|jjdk(ro|jdddfj}|jdddfj}||z |jdddf<||z |jdddf<n'||jdddfz |jdddf<||jdz |jd<|j ||j dz |j d<yy)zLFlips the coordinates of bounding boxes, segments, and keypoints vertically.rNrr1rsrcr*r)copyrird)r+r{y1y2s rflipudzInstances.flipud8 <<  & (QT"'')BQT"'')B !BDKK1  !BDKK1  !DKK1$5 5DKK1  !DMM&$9 9 f >> %%&)?%?DNN6 " &rc|jjdk(ro|jdddfj}|jdddfj}||z |jdddf<||z |jdddf<n'||jdddfz |jdddf<||jdz |jd<|j ||j dz |j d<yy)zCReverses the order of the bounding boxes and segments horizontally.rNrrrrr)r+rzx1x2s rfliplrzInstances.fliplrErrc|jj}|jd|jddddgfj d||jddddgf<|jddddgfj d||jddddgf<|dk7r|j||j dj d||j d<|j d j d||j d <|j Y|j dj d||j d<|j d j d||j d <yy) zUClips bounding boxes, segments, and keypoints values to stay within image boundaries.rrlNrrrr1rrrs)rcr*rmr)cliprird)r+rzr{ ori_formats rrzInstances.clipRsB\\((  (!%QAY!7!> %%)^^F%;%@%@A%FDNN6 "%)^^F%;%@%@A%FDNN6 " &rc|jdkD}t|s]|j||_t|jr|j||_|j |j ||_|S)z Remove zero-area boxes, i.e. after clipping some boxes may have zero width or height. This removes them. r)rorNrcr8rird)r+goods rremove_zero_area_boxesz Instances.remove_zero_area_boxes`sd "4y<<-DL4==! $ d 3 ~~)!%!5 rcvt||jj|_|||_|||_yy)zUpdates instance variables.rlN)r rcr*rird)r+r)rirds rupdatezInstances.updateos9fT\\-@-@A  $DM  &DN !rc,t|jS)z'Return the length of the instance list.r?r2s rr@zInstances.__len__wrArinstances_listct|ttfsJ|s|tjdSt d|DsJt |dk(r|dS|djdu}|djj}|dj}tj|Dcgc]}|jc}|}tj|Dcgc]}|jc}|} |r/tj|Dcgc]}|jc}|nd} ||| | ||Scc}wcc}wcc}w)a Concatenates a list of Instances objects into a single Instances object. Args: instances_list (List[Instances]): A list of Instances objects to concatenate. axis (int, optional): The axis along which the arrays will be concatenated. Defaults to 0. Returns: Instances: A new Instances object containing the concatenated bounding boxes, segments, and keypoints if present. Note: The `Instances` objects in the list should have the same properties, such as the format of the bounding boxes, whether keypoints are present, and if the coordinates are normalized. rc3<K|]}t|tywrE)rr`)rFinstances rrHz(Instances.concatenate..sR>x:h 2>rIrNrJ)rr7rrLrMrNr8rdrcr*rerOr)ri) rPrrK use_keypointrjreins cat_boxesrQ cat_segments cat_keypointss rrOzInstances.concatenate{s%$.4-888rxx{# #R>RRRR ~ ! #!!$ $%a(22$> $Q'//66 #A&11 NN.#I.3CJJ.#IPTU ~~>&J>aqzz>&JQUV \h^'L^ ^'LSWXnr 9lM; SS$J&J'Ls)D3D8 D=c.|jjS)zReturn bounding boxes.)rcr)r2s rr)zInstances.bboxess||"""r)NNrTrX)F)r"r`)NNrY)rZr[r\r]r,rmpropertyror9r|r~rrWrrrrrr@r^rrOr)rrrr`r`s>!,,$$ .   + 8 @ @ G '  Tk): T{ T TD##rr`) collectionsr itertoolsrnumbersrtypingrnumpyrLopsrr r r r r rr to_2tupler6r&__all__r r`rrrrs]ddd AJ AJ $ KK\q#q#r