a yfy?@sddlmZddlmZddlmZddlmZddlZ ddl m Z m Z m Z mZmZmZdd Zed Zed Zgd Zd ZGdddZGdddZdS))abc)repeat)Number)ListN) ltwh2xywh ltwh2xyxy xywh2ltwh xywh2xyxy xyxy2ltwh xyxy2xywhcsfdd}|S)zFrom PyTorch internals.cst|tjr|Stt|S)z2Parse bounding boxes format between XYWH and LTWH.) isinstancerIterabletupler)xnV/var/www/html/django/DPS/env/lib/python3.9/site-packages/ultralytics/utils/instance.pyparsesz_ntuple..parser)rrrrr_ntuple s r)xyxyxywhZltwh)Bboxesc@speZdZdZdddddZddZd d Zd d Zd dZddZ e de dddddZ ddddZ dS)ra 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. rNreturncCsj|tvsJd|dt|jdkr6|dddfn|}|jdksHJ|jddksZJ||_||_dS)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$rrr__init__1s zBboxes.__init__cCs|tvsJd|dt|j|kr*dS|jdkrF|dkr@tnt}n,|jdkrb|dkr\tnt}n|dkrntnt}||j|_||_dS)z6Converts bounding box format from one type to another.rrNrr) r r$r r r r rrr#)r%r$funcrrrconvert;s    zBboxes.convertcCsv|jdkrR|jdddf|jdddf|jdddf|jdddfS|jdddf|jdddfS)zReturn box areas.rNrrr)r$r#r%rrrareasIsJ"z Bboxes.areascCst|trt|}t|ttfs$Jt|dks4J|jdddf|d9<|jdddf|d9<|jdddf|d9<|jdddf|d9<dS)z Multiply bounding box coordinates by scale factor(s). Args: scale (int | tuple | list): Scale factor(s) for four coordinates. If int, the same scale is applied to all coordinates. rNrrrr)r r to_4tuplerlistlenr#)r%scalerrrmulas z Bboxes.mulcCst|trt|}t|ttfs$Jt|dks4J|jdddf|d7<|jdddf|d7<|jdddf|d7<|jdddf|d7<dS)z Add offset to bounding box coordinates. Args: offset (int | tuple | list): Offset(s) for four coordinates. If int, the same offset is applied to all coordinates. rNrrrr)r,)r%offsetrrraddrs z Bboxes.addcCs t|jS)zReturn the number of boxes.r/r#r*rrr__len__szBboxes.__len__r) boxes_listrcCsjt|ttfsJ|s$|tdStdd|Ds:Jt|dkrN|dS|tjdd|D|dS)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. rcss|]}t|tVqdSN)r r).0boxrrr z%Bboxes.concatenate..rcSsg|] }|jqSrr#r8brrr r;z&Bboxes.concatenate..axis)r r.rnpemptyallr/ concatenate)clsr6rArrrrEs zBboxes.concatenatecCsLt|tr t|j|ddS|j|}|jdksDJd|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!)r intrr#viewr!)r%indexr>rrr __getitem__s   zBboxes.__getitem__)r)r)__name__ __module__ __qualname____doc__r&r(r+r1r3r5 classmethodrrErKrrrrr"s rc@seZdZdZd)ddddZdd Zed d Zd*d dZddZ ddZ ddZ ddddZ ddZ ddZddZddZd+dd Zd!d"Zed,eddd$d%d&Zed'd(ZdS)- 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. NrTrcCs$t||d|_||_||_||_dS)a' Initialize the object with bounding boxes, segments, and keypoints. Args: bboxes (np.ndarray): Bounding boxes, shape [N, 4]. segments (list | np.ndarray, optional): Segmentation masks. Defaults to None. keypoints (np.ndarray, optional): Keypoints, shape [N, 17, 3] and format (x, y, visible). Defaults to None. bbox_format (str, optional): Format of bboxes. Defaults to "xywh". normalized (bool, optional): Whether the coordinates are normalized. Defaults to True. )r#r$N)r_bboxes keypoints normalizedsegments)r%r#rUrS bbox_formatrTrrrr&s zInstances.__init__cCs|jj|ddS)zConvert bounding box format.r$N)rRr()r%r$rrr convert_bboxszInstances.convert_bboxcCs |jS)z%Calculate the area of bounding boxes.)rRr+r*rrr bbox_areasszInstances.bbox_areasFcCst|jj||||fd|rdS|jd|9<|jd|9<|jdurp|jd|9<|jd|9<dS)z8Similar to denormalize func but without normalized sign.r0N.r.r)rRr1rUrS)r%Zscale_wZscale_hZ bbox_onlyrrrr0s zInstances.scalecCs||js dS|jj||||fd|jd|9<|jd|9<|jdurr|jd|9<|jd|9<d|_dS)zHDenormalizes boxes, segments, and keypoints from normalized coordinates.NrZr[r\FrTrRr1rUrSr%whrrr denormalizes zInstances.denormalizecCs|jr dS|jjd|d|d|d|fd|jd|<|jd|<|jdur|jd|<|jd|<d|_dS)zFNormalize bounding boxes, segments, and keypoints to image dimensions.NrrZr[r\Tr]r^rrr normalize s& zInstances.normalizecCsz|jrJd|jj||||fd|jd|7<|jd|7<|jdurv|jd|7<|jd|7<dS)z!Handle rect and mosaic situation.z1you should add padding with absolute coordinates.)r2r[r\N)rTrRr3rUrS)r%ZpadwZpadhrrr add_paddings zInstances.add_paddingcCsXt|jr|j|n|j}|jdur.|j|nd}|j|}|jj}t|||||jdS)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#rUrSrVrT)r/rUrSr#rRr$rQrT)r%rJrUrSr#rVrrrrKs zInstances.__getitem__cCs|jjdkrf|jdddf}|jdddf}|||jdddf<|||jdddf<n$||jdddf|jdddf<||jd|jd<|jdur||jd|jd<dS)zLFlips the coordinates of bounding boxes, segments, and keypoints vertically.rNrr)r\rRr$r#copyrUrS)r%r`y1y2rrrflipud;s $ zInstances.flipudcCs|jjdkrf|jdddf}|jdddf}|||jdddf<|||jdddf<n$||jdddf|jdddf<||jd|jd<|jdur||jd|jd<dS)zCReverses the order of the bounding boxes and segments horizontally.rNrrr[rd)r%r_x1Zx2rrrfliplrHs $ zInstances.fliplrcCs|jj}|jdd|jddddgfd||jddddgf<|jddddgfd||jddddgf<|dkr|j|d|jdd||jd<|jd d||jd <|jdur|jdd||jd<|jd d||jd <dS) zUClips bounding boxes, segments, and keypoints values to stay within image boundaries.rrWNrrrr)r[r\)rRr$rXr#cliprUrS)r%r_r`Z ori_formatrrrrkUs 00  zInstances.clipcCsN|jdk}t|sJ|j||_t|jr4|j||_|jdurJ|j||_|S)zURemove zero-area boxes, i.e. after clipping some boxes may have zero width or height.rN)rYrDrRr/rUrS)r%Zgoodrrrremove_zero_area_boxescs      z Instances.remove_zero_area_boxescCs2t||jjd|_|dur ||_|dur.||_dS)zUpdates instance variables.rWN)rrRr$rUrS)r%r#rUrSrrrupdatens zInstances.updatecCs t|jS)z'Return the length of the instance list.r4r*rrrr5vszInstances.__len__r)instances_listrc Cst|ttfsJ|s$|tdStdd|Ds:Jt|dkrN|dS|djdu}|djj }|dj }tj dd|D|d}tj d d|D|d}|rtj d d|D|dnd}||||||S) 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. rcss|]}t|tVqdSr7)r rQ)r8instancerrrr:r;z(Instances.concatenate..rNcSsg|] }|jqSrr<)r8Zinsrrrr?r;z)Instances.concatenate..r@cSsg|] }|jqSr)rUr=rrrr?r;cSsg|] }|jqSr)rSr=rrrr?r;) r r.rrBrCrDr/rSrRr$rTrE) rFrnrAZ use_keypointrVrTZ cat_boxesZ cat_segmentsZ cat_keypointsrrrrEzs    zInstances.concatenatecCs|jjS)zReturn bounding boxes.)rRr#r*rrrr#szInstances.bboxes)NNrT)F)NN)r)rLrMrNrOr&rXpropertyrYr0rarbrcrKrhrjrkrlrmr5rPrrEr#rrrrrQs(        "rQ) collectionsr itertoolsrnumbersrtypingrnumpyrBopsrrr r r r rZ to_2tupler-r __all__rrQrrrrs