o jĎiT @sddlmZmZddlmZmZddlmZddlm Z     ddede d e d ee e fd e d eeeff d dZ ddedede fddZddede fddZddedede fddZddedede fddZdS))TupleUnion)BboxTypeGenericBboxType)BaseBoundingBox load_bboxNTbbox from_typeto_type image_size return_valuesreturncKsPt|ts|s tdtd|||d|}t|d|}|}|r&|jS|S)a Converts given bbox with given `from_type` to given `to_type`. It uses VOC format as an intermediate format. Args: bbox: (generic) Bounding box. from_type: (str) Type/Format of the given bounding box. to_type: (str) Type/Format of the resulting bounding box. image_size: (tuple(int,int)) Image size as (w, h) tuple, it is required if the one side of the boxes requires scaling. return_values: (bool) Whether to return values as a Tuple, or BoundingBox object. True by default for compatibility purposes. Return: Bounding box in type `to_type`. z?if `bbox` is not a BoundingBox object, `from_type` is required.)namevaluesr to_N) isinstancer ValueErrorrgetattrr)r r r r r kwargssource_to_target target_bboxrrO/home/jeff/fluffinator/venv/lib/python3.10/site-packages/pybboxes/functional.py convert_bboxs rcocobbox1bbox2 bbox_typecKs0td||d|}td||d|}||S)a9 Computes intersection area between given bounding boxes. Args: bbox1: Bounding box 1. bbox2: Bounding box 2. bbox_type: Format of the bounding boxes. It's 'coco' [x-tl, y-tl, w, h] by default. Returns: Intersection area if bounding boxes intersect, 0 otherwise. )rrNrrrrrrrrrcompute_intersection+s r cKst|||fi|S)z2 Computes the area of given bounding box. )r )r rrrrr compute_area<sr!cKsJt||fd|i|}t|fd|i|}t|fd|i|}|||S)z Computes union area of given boxes. Args: bbox1: Bounding box 1. bbox2: Bounding box 2. bbox_type: Format of the bounding boxes. It's 'coco' [x-tl, y-tl, w, h] by default. Returns: Union area. r)r r!)rrrr intersectionarea1area2rrr compute_unionCs  r%cKs(t|||fi|t|||fi|S)a2 Computes Intersection over Union (IoU) (special form of Jaccard Index) metric. Args: bbox1: Bounding box 1. bbox2: Bounding box 2. bbox_type: Format of the bounding boxes. It's 'coco' [x-tl, y-tl, w, h] by default. Returns: Intersection over Union ratio. )r r%rrrr compute_iouUs( r&)NNNT)r)typingrrpybboxes._typingrrpybboxes.boxes.baserpybboxes.boxes.bboxrstrintboolrr r!r%r&rrrrs2     #