o jĎi@slddlZddlmZmZddlmZmZmZddlZ ddl m Z gdZ GdddZ Gdd d e eZdS) N)ABCabstractmethod)ListTupleUnion)find_stack_level)albumentationsfiftyoneyoloc@seZdZdedededefddZdddefd d Zdddefd d Zdddefd dZdddefddZ e defddZ e defddZ e defddZ dddefddZdddefddZdddefddZdS)Boxx_tly_tlx_bry_brcCs||_||_||_||_dSN)r r rr)selfr r rrrO/home/jeff/fluffinator/venv/lib/python3.10/site-packages/pybboxes/boxes/base.py__init__ s z Box.__init__otherreturncC ||Sr)unionrrrrr__add__ z Box.__add__cCst|j||Srintarea intersectionrrrr__sub__z Box.__sub__cCrr)rrrrr__mul__rz Box.__mul__cCs |j|jSr)rrrrr __truediv__s zBox.__truediv__cCs |j|jSr)widthheightrrrrrs zBox.areacCt|j|jSr)rrr r&rrrr%#z Box.heightcCr'r)rrr r&rrrr$'r(z Box.widthcCspt|j|jf|j|jf\}}t|j|jf|j|jf\}}||ks(||kr*dS||}||}t||S)Nr)npmaximumr r minimumrrr)rrr r rrintersection_widthintersection_heightrrrr+s   zBox.intersectioncCst|j|j||Srrrrrrr4sz Box.unioncCs||||Sr)rrrrrriou7r!zBox.iouN)__name__ __module__ __qualname__rrrr r"floatr#propertyrr%r$rrr.rrrrr s r csfeZdZ  dAdeeefdeeefdeeefdeeefdeeefdef fd d Zd d Z e d eedffddZ e ddZ e j deeeffddZ ddZe d efddZd efddZeddZddZdBddZdCded eeeeeefdffd d!ZdCded eeeeeefdffd"d#ZdCded eeeeeefdffd$d%ZedCded eeeeeefdffd&d'ZdCded eeeeeefdffd(d)Ze d*d+Zd,ed dfd-d.ZdDd/d0Zd1ed dfd2d3Zd4ed dfd5d6Ze e  7dEd8ed9ed:ed;edeeefded dfdeee"e#j$ffd?d@Z%Z&S)FBaseBoundingBoxNFv1v2v3v4 image_sizestrictcsB||_||_d|_||||||jdd}tt|j|dS)NT) return_values) _image_sizer:_is_oob_validate_and_set_valuesto_vocsuperr4r)rr5r6r7r8r9r: voc_values __class__rrr<s   zBaseBoundingBox.__init__c CsP|j\}}ddd|jD}d|d|jd|jd|pdd|p$dd S) N cSs&g|]}t|tr |dnt|qS)z.4f) isinstancer2str).0vrrr Ns&z,BaseBoundingBox.__repr__..z<[z] (xz ) | Image: (?z)>)r9joinvaluesr$r%)r image_width image_heightstr_valsrrr__repr__Ls 0zBaseBoundingBox.__repr__rcC|jS)z Whether the box is OOB (Out-of-bounds). Returns: None -> unknown. False -> Not OOB. True -> OOB. )r=r&rrris_oobQszBaseBoundingBox.is_oobcCs|jdur|jSdS)NNNr<r&rrrr9[s zBaseBoundingBox.image_sizecCs ||_dSrrU)rr9rrrr9bs cCs|jdkrdSdS)NrTTF)r9r&rrris_image_size_nullfs z"BaseBoundingBox.is_image_size_nullcCrRr_valuesr&rrrrMkszBaseBoundingBox.valuescGs|SrrrrMrrr_correct_value_typesosz$BaseBoundingBox._correct_value_typescGdSrrrYrrr_validate_valuesrz BaseBoundingBox._validate_valuescGs ||_dS)g This method is intended to be "final", and should not be overridden in child classes. NrWrYrrr _set_valuesvs zBaseBoundingBox._set_valuescGs(||_|j|}|j||j|dS)zM Validate and sets given values if validation is successful. N) raw_valuesrZr\r_rYrrrr>|s  z(BaseBoundingBox._validate_and_set_valuesr;cC||Sr)r?to_albumentationsrr;rrrrbz!BaseBoundingBox.to_albumentationscCrar)r?to_cocorcrrrrerdzBaseBoundingBox.to_cococCrar)r? to_fiftyonercrrrrfrdzBaseBoundingBox.to_fiftyonecCr[rrrcrrrr?r]zBaseBoundingBox.to_voccCrar)r?to_yolorcrrrrgrdzBaseBoundingBox.to_yolocCs|jjddS)N boundingbox)rCr/lowerreplacer&rrrnameszBaseBoundingBox.nameopcOsT|}t||}||i|}t|d|j}|}|j|j|j|jddS)Nto_)r9r:)r?getattrrlrrMr9r:)rrmargskwargs refined_boxbox_opbox_conversionrrr_generic_operations  z"BaseBoundingBox._generic_operationcCs|d|S)zh Clamps the box with respect to the image borders. If the box is not OOB, does nothing. clamprur&rrrrvs zBaseBoundingBox.clampfactorcCs|d||S)Nscalerw)rrxrrrrys zBaseBoundingBox.scaleamountcCs<|jtvr|j\}}|d||d|f}|d||S)a( Perform a shift operation on the bounding box inplace. Args: amount: The amount to shift the bounding box. The first value is the amount to shift the x-coordinate, and the second value is the amount to shift the y-coordinate. rshift)rlNORMALIZED_BOXESr9ru)rrzr$r%rrrr|s  zBaseBoundingBox.shiftTr r rrcCr[rr)clsr r rrr9r:rrrfrom_vocs zBaseBoundingBox.from_vocarcKs@tjdttdt|dkrtdt|d||i|S)r^zThe functionality of the `from_array()` method is changed from only supporting a single box values to support (arbitrary) n-dimensional array of box values starting from 0.2 onward requiring Python3.8 or higher.) stacklevelz,Given array must be length of 4, got length .)warningswarn FutureWarningrlen ValueError)r~rrqrrr from_arrays zBaseBoundingBox.from_array)NF)rN)F)rr4)NT)'r/r0r1rrr2rboolrrQr3rSr9setterrVrMrZrr\r_r>rbrerfr?rgrlrFrurvryr| classmethodrrr)ndarrayr __classcell__rrrBrr4;s        ( ((*(    $r4)rabcrrtypingrrrnumpyr)pybboxes.utilsrr}r r4rrrrs /