jH ddlmZddlmZddlZddlZddlmZm Z gdZ dede fd Z d e defd Z d e ejdee eje ejffd Zde eejzdeejzfdZd2de ejdedejfdZd3dejdedejfdZdeeefdedeeeffdZ d4dejdedeeefdedejf dZdejdeeefd eeefdejfd!Z d5dejdedeeefd#ed$eeefdzdejf d%Z d6d'ejdeded(edejf d)Zd'ejdejfd*Zd+ejdefd,Zdejdeeefdejfd-Zd.ejd ejde ejfd/Zej d&fd.ejd+ejd0ede ejfd1Z!dS)7)deepcopy)ceilN) BoundingBox Polygon4P)bbox_to_polygonpolygon_to_bboxresolve_enclosing_bboxresolve_enclosing_rbbox rotate_boxescompute_expanded_shape rotate_imageremove_image_paddingestimate_page_angleconvert_to_relative_coordsrotate_abs_geoms extract_cropsextract_rcrops detach_scoresbboxreturnc|d|dd|ddf|dd|ddf|dfS)zpConvert a bounding box to a polygon Args: bbox: a bounding box Returns: a polygon rr)rs W/var/www/html/Carbon-Document/venv/lib/python3.11/site-packages/doctr/utils/geometry.pyrr sE 7T!WQZa,tAwqz471:.FQ OOpolygonct|\}}t|t|ft|t|ffS)zsConvert a polygon to a bounding box Args: polygon: a polygon Returns: a bounding box )zipminmax)rxys rr r ,s> =DAq FFCFF c!ffc!ff- --rboxescdtjdttjtjffdtfd|D\}}t |t |fS)a)Detach the objectness scores from box predictions Args: boxes: list of arrays with boxes of shape (N, 5) or (N, 5, 2) Returns: a tuple of two lists: the first one contains the boxes without the objectness scores, the second one contains the objectness scores r#rc|jdkr|ddddf|dddffS|ddddf|ddddffS)N)ndim)r#s r_detachzdetach_scores.._detachDs^ :??CRC=%2,. .QQQV}eAAAr2I...rc3.K|]}|VdS)Nr).0boxr)s r z detach_scores..Is+!@!@3''#,,!@!@!@!@!@!@r)npndarraytuplerlist)r# loc_preds obj_scoresr)s @rrr9st/rz/eBJ ,B&C////  !@!@!@!@%!@!@!@AIz  ??D,, ,,rbboxesct|tjr|tj|dd\}}}}tj||||gStd|D\}}t |t |ft |t |ffS)aGCompute enclosing bbox either from: Args: bboxes: boxes in one of the following formats: - an array of boxes: (*, 4), where boxes have this shape: (xmin, ymin, xmax, ymax) - a list of BoundingBox Returns: a (1, 4) array (enclosing boxarray), or a BoundingBox raxiscg|] }|D]}| Srr)r+r,points r z*resolve_enclosing_bbox.._s%>>>s#>>U>>>>r) isinstancer.r/splitarrayrr r)r4xminyminxmaxymaxr!r"s rr r Ms&"*%%2!#&!!!>f>>>?1AA#a&&#a&&!111rrbboxes intermed_sizectj|d}||z}tj|tj}tj||z S)aCompute enclosing rotated bbox either from: Args: rbboxes: boxes in one of the following formats: - an array of boxes: (*, 4, 2), where boxes have this shape: (x1, y1), (x2, y2), (x3, y3), (x4, y4) - a list of BoundingBox intermed_size: size of the intermediate image Returns: a (4, 2) array (enclosing rotated box) rr7)r. concatenatecv2 minAreaRectastypeint32 boxPoints)rDrEcloudrects rr r csTwQ777E ]E ?5<<11 2 2D =   ..rpointsanglec.|tjzdz }tjtj|tj| gtj|tj|gg|j}tj||jS)zRotate points counter-clockwise. Args: points: array of size (N, 2) angle: angle between -90 and +90 degrees Returns: Rotated points f@dtype)r.pir>cossinrUmatmulT)rPrQ angle_rad rotation_mats rrotate_abs_pointsr]ys %I8 &  bfY/// 026)3D3DbfYFWFW2XYagamL 9V\^ , ,,r img_shapectj|ddz |ddz g|d dz |ddz gg}t||}dtj|dz}|d|dfS)zCompute the shape of an expanded rotated image Args: img_shape: the height and width of the image angle: angle between -90 and +90 degrees Returns: the height and width of the rotated image rr&rr7)r.r>r]absr )r^rQrProtated_pointswh_shapes rr r s 1 9Q min_angle, around the center of the page. If target_shape is specified, the boxes are remapped to the target shape after the rotation. This is done to remove the padding that is created by rotate_page(expand=True) Args: loc_preds: (N, 4) or (N, 4, 2) array of RELATIVE boxes angle: angle between -90 and +90 degrees orig_shape: shape of the origin image min_angle: minimum angle to rotate boxes target_shape: shape of the destination image Returns: A batch of rotated boxes (N, 4, 2): or a batch of straight bounding boxes r&Nrrrfr7ZrSrTr')rprq) rur(r.rir`rVr>rWrXrUrYr{) r2rQrpr}ro_boxesr[r\rP image_centerra rotated_boxess rr r s/.^^  F {aqqq1a&y!qqq1a&y!qqq1a&y!qqq1a&y!       5zzIUb9n!9::;; < < AtIaL5;q>9::;; < <&%1*eeqj.@!AEQJPUX]abXbPbCcek lmmM"1"%MFE%uqy&1*&=ucJJGnWgv??G b KNU[^ +q1AGMRSDT1T U U a 7=#33 AUV8WXX #gmA&6Q&G%+VW.&X[b[hij[k&k"l"lu #7=#3ek!n#Du{ST~#UX_XefgXh#hiiklufW EEQJ4F'G%ST*V[^cgh^hVhIikq&rssG  bj%+crc*:44R4*@PSP`aaaG Nrctj|d}tj|d}tj|dddg\}}tj|dddg\}}|||dz||dzfS)zRemove black border padding from an image Args: image: numpy tensor to remove padding from Returns: Image with padding removed rr7rr')r.anywhere)rrowscolsrminrmaxcmincmaxs rrrLs 6%a D 6%a D$"Ar7+JD$$"Ar7+JD$ q$/1 22rrmc  |ddddf|ddddfz}|ddddf|ddddfz}|ddddf|ddddfz}|ddddf|ddddfz}tjdd5 ttjtj||z ||z z dztjz cdddS#t $rYdddd SwxYw#1swxYwYdS) zTakes a batch of rotated previously ORIENTED polys (N, 4, 2) (rectified by the classifier) and return the estimated angle ccw in degrees Nrrfrr&raise)divideinvalidrO)r.errstatefloatmedianarctanrVFloatingPointError)rmxleftyleftxrightyrights rrr^s !!!Q'NU111a7^ +E !!!Q'NU111a7^ +E 111a7^eAAAq!Gn ,F 111a7^eAAAq!Gn ,F GW 5 5 5  ")UV^$GHH3NQSQVVWW "       s1C9A C C6'C95C66C99C=C=ch|jdkrv|jdddkrctj|jtj}|d|dz |d<|d|dz |d<|ddS|jd kr|jdd krtj|jtj}|ddddd f|dz |ddddd f<|ddddd f|dz |ddddd f<|ddSt d |j) zConvert a geometry to relative coordinates Args: geoms: a set of polygons of shape (N, 4, 2) or of straight boxes of shape (N, 4) img_shape: the height and width of the image Returns: the updated geometry rfrNr6r&rTrgrhrr&r6z invalid format for arg `geoms`: )r(rr.emptyrjrlrt)rcr^polygonsr#s rrrpsA zQ5;qrr?f44!x 2:FFF =9Q<7 =9Q<7}}Q""" zQ5;q>Q..HU[ CCCaaa1f ! 4aaa1f qqq!$Q$w)A,6aaaAgzz!Q E EE F FFrimgc|jddkrgS|jddkrtd|}jdd\}}tj|jtjsj|ddddgfxx|zcc<|ddddgfxx|zcc<|t}|ddxxdz cc<tfd|DS) aCreated cropped images from list of bounding boxes Args: img: input image boxes: bounding boxes of shape (N, 4) where N is the number of boxes, and the relative coordinates (xmin, ymin, xmax, ymax) Returns: list of cropped images rrr6zGboxes are expected to be relative and in order (xmin, ymin, xmax, ymax)Nr&rfc\g|](}|d|d|d|df)S)rrfrr&r)r+r,rs rr;z!extract_crops..s;MMMsSQ#a&#a&3q6/9:MMMr) rAssertionErrorrur. issubdtyperUintegerroundrJrr)rr#rhws` rrrs {1~  {1~fgggZZ\\F 9RaR=DAq =rz 2 2qqq1a&yQqqq1a&yQ&&s++qrr a MMMMfMMM N NNrassume_horizontalc |jddkrgS|jdddkrtd|}|jdd\}}tj|jtjs.|dddddfxx|zcc<|dddddfxx|zcc<||r'g}|D]}tj|d} ||dddf| dk} ||dddf| dk} | tj| dddf} | d} | d} | tj| dddf} | d}| d}tj | | ||g| }tj || z }tj || z }tj | | z }tj ||z }tt|t|}tt|t|}tj ddgd|dz g|dz dg|dz |dz gg| tj|}tj|||f}||!n |dddd ftjtj dddfdddfz dtj dddfdddfz dtj|jdd df| dz xddddf<ddddf<dz ddddf<fd t+|jdD}|S) aMCreated cropped images from list of rotated bounding boxes Args: img: input image polys: bounding boxes of shape (N, 4, 2) dtype: target data type of bounding boxes assume_horizontal: whether the boxes are assumed to be only horizontally oriented Returns: list of cropped images rrNrz:polys are expected to be quadrilateral, of shape (N, 4, 2)r&r7r'rTrfc g|]_}tjtj||t|t|f`Sr)rHrgetAffineTransformr)r+idxd1d2dst_ptssrc_imgsrc_ptss rr;z"extract_rcrops.. sn    N&ws|WS\BBRWs2c7||,      r)rrrur.rrUrmeanargsortr>linalgnormr rrHgetPerspectiveTransformwarpPerspectiveappendrJrjzerosrange)rrmrUrrrrcropsr,centroid left_points right_points top_left_ptbottom_left_pt top_right_ptbottom_right_pt box_points width_upper width_lower height_left height_right rect_width rect_height affine_matcroprrrrrs @@@@@rrrsW {1~  {122&  YZZZZZ\\FIbqbMMFE =rz 2 2"qqq!!!Qw5 qqq!!!Qw6!GO 7 7 Cws+++Hc!!!Q$i(1+56Ks111a4yHQK78L&bjQQQT1B&C&CDK%a.K(_N' <13E(F(FGL'?L*2.OnlOLJ ).. )CDDK)..>)IJJK)..+)EFFK9>>/L*HIILS--s;/?/?@@Jc+..L0A0ABBKhF a(!^Q'!^[1_5   G4ZIIJ&[)D LL    o7 vBQB-&&rz22 Y^^GAAAqDMGAAAqDM9^ C C Y^^GAAAqDMGAAAqDM9^ C C(FLOQ2%@@@.01f41a7111a7+61a        V\!_--    Lr)rC)rO)T)r|N)FF)"rurmathrrHnumpyr. common_typesrr__all__rr r1r/r0rr rr rr]r boolrr{r rrrrrrjrrrrrsa  00000000   $ P+ P) P P P P .Y .; . . . .-bj)-eD4Dd2:FV4V.W----(24 #4rz#A2kTVT^F^2222,//T"*%5/c/UWU_////,--bj------"$eCHo$e$cSVh$$$$2 && :& &S#X&  & Z &&&&R2:5c?PUVY[^V^P_dfdn<+/ 66z6 6c3h6 6 S/D( 6 Z 6666x"' .. :. . . . Z ....b3 3rz3333$rze$GbjGU38_GQSQ[GGGG2OrzO"*Obj9IOOOO</1jTYmm mJmMQm "*mmmmmmr