o Ďi^@sddlmZddlZddlmZddlZddlmZm Z ddl m Z m Z m Z ddlmZeddGd d d ZeddGd d d ZGd ddZGdddZdS)) annotationsN) dataclass)CocoAnnotationCocoPrediction)get_bbox_from_coco_segmentation$get_bool_mask_from_coco_segmentation$get_coco_segmentation_from_bool_mask)ShapelyAnnotationT)frozenc@seZdZUdZded<dZded<ddZed d Zed d Z ed dZ eddZ eddZ eddZ eddZd,d-ddZd d!Zd"d#Zd$d%Zd&d'Zd(d)Zd*d+ZdS). BoundingBoxaBoundingBox represents a rectangular region in 2D space, typically used for object detection annotations. Attributes: box (Tuple[float, float, float, float]): The bounding box coordinates in the format (minx, miny, maxx, maxy). - minx (float): Minimum x-coordinate (left). - miny (float): Minimum y-coordinate (top). - maxx (float): Maximum x-coordinate (right). - maxy (float): Maximum y-coordinate (bottom). shift_amount (Tuple[int, int], optional): The amount to shift the bounding box in the x and y directions. Defaults to (0, 0). !!! example "BoundingBox Usage Example" ```python bbox = BoundingBox((10.0, 20.0, 50.0, 80.0)) area = bbox.area expanded_bbox = bbox.get_expanded_box(ratio=0.2) shifted_bbox = bbox.get_shifted_box() coco_format = bbox.to_coco_bbox() ``` z/tuple[float, float, float, float] | list[float]box)rrztuple[int, int] shift_amountcCsDt|jdkstdd|jDrtdt|jdkr tddS)Ncss|]}|dkVqdS)rN).0coordrrK/home/jeff/fluffinator/venv/lib/python3.10/site-packages/sahi/annotation.py ,sz,BoundingBox.__post_init__..z;box must be 4 non-negative floats: [minx, miny, maxx, maxy]z3shift_amount must be 2 integers: [shift_x, shift_y])lenr any ValueErrorr selfrrr __post_init__+s "zBoundingBox.__post_init__cC |jdSNrr rrrrminx1 zBoundingBox.minxcCrNrrrrrminy5rzBoundingBox.minycCr)Nrrrrrrmaxx9rzBoundingBox.maxxcCr)Nrrrrrmaxy=rzBoundingBox.maxycCrrr rrrrshift_xArzBoundingBox.shift_xcCrr r&rrrrshift_yErzBoundingBox.shift_ycCs|j|j|j|jS)N)r#rr%r"rrrrareaIszBoundingBox.area皙?Nratiofloatmax_x int | Nonemax_yc Cs|j|j}|j|j}t||}t||}|r"t||j|n|j|}td|j|} |r9t||j|n|j|} td|j|} | | || g} t| S)aReturns an expanded bounding box by increasing its size by a given ratio. The expansion is applied equally in all directions. Optionally, the expanded box can be clipped to maximum x and y boundaries. Args: ratio (float, optional): The proportion by which to expand the box size. Default is 0.1 (10%). max_x (int, optional): The maximum allowed x-coordinate for the expanded box. If None, no maximum is applied. max_y (int, optional): The maximum allowed y-coordinate for the expanded box. If None, no maximum is applied. Returns: BoundingBox: A new BoundingBox instance representing the expanded box. r)r#rr%r"intminmaxr ) rr+r-r/why_marx_marr#rr%r"r rrrget_expanded_boxMs     zBoundingBox.get_expanded_boxcCs |j|j|j|j|j|jgS)zReturns [xmin, ymin, width, height] Returns: List[float]: A list containing the bounding box in the format [xmin, ymin, width, height]. rr"r#r%rrrrto_xywhhs zBoundingBox.to_xywhcC|S)z Returns the bounding box in COCO format: [xmin, ymin, width, height] Returns: List[float]: A list containing the bounding box in COCO format. )r9rrrr to_coco_bboxqzBoundingBox.to_coco_bboxcCs|j|j|j|jgS)z Returns: [xmin, ymin, xmax, ymax] Returns: List[float]: A list containing the bounding box in the format [xmin, ymin, xmax, ymax]. r8rrrrto_xyxyzszBoundingBox.to_xyxycCr:)z Returns the bounding box in VOC format: [xmin, ymin, xmax, ymax] Returns: List[float]: A list containing the bounding box in VOC format. )r=rrrr to_voc_bboxr<zBoundingBox.to_voc_bboxcCs4|j|j|j|j|j|j|j|jg}t|S)zReturns shifted BoundingBox. Returns: BoundingBox: A new BoundingBox instance representing the shifted box. )rr'r"r(r#r%r )rr rrrget_shifted_boxs    zBoundingBox.get_shifted_boxcCs8d|j|j|j|jfd|j|jd|j|jdS)NzBoundingBox: r8rrrr__repr__s   zBoundingBox.__repr__)r*NN)r+r,r-r.r/r.)__name__ __module__ __qualname____doc____annotations__r rpropertyrr"r#r%r'r(r)r7r9r;r=r>r?rArrrrr s4              r c@s2eZdZUdZded<ded<ddZdd Zd S) CategoryzCategory of the annotation. Attributes: id (int): Unique identifier for the category. name (str): Name of the category. r0idstrnamecCs,t|jts tdt|jtstddS)Nzid should be integerzname should be string) isinstancerIr0 TypeErrorrKrJrrrrrs  zCategory.__post_init__cCsd|jd|jdS)NzCategory: $"z)Mask.get_shifted_mask..rr!rcrgr)r1r(rUrhrjrrrl%rmcSsg|] }|D]}|qqSrr)rrijrrrrl&sr[) rUrVrrPrangerappendziprOrR)rshifted_segmentationxsysrrjrget_shifted_masks $ zMask.get_shifted_maskN)rPrQrRrSr rT)rXrYrRrSrZr,r rT)r^rYrRrSr rT)rarY)rarS)rarO)rBrCrDrErW classmethodr_r`rGr^rdrRr rurrrrrOs(     rOc@seZdZdZddddddgdfd4ddZeddddgdfd5ddZeddddgfd6ddZeddddgdfd7ddZedddgfd8ddZ eddddgfd9ddZ eddgdfd:dd Z d;d#d$Z dObjectAnnotationzs   w