j=rddlZddlZddlZddlmZddlmZdgZGddeZ GddZ dS) N)Polygon)DetectionPostProcessorDBPostProcessorc eZdZdZ ddedededd ffd Zd ejdejfd Z d ejdejdejfdZ xZ S)raImplements a post processor for DBNet adapted from the implementation of `xuannianz `_. Args: unclip ratio: ratio used to unshrink polygons min_size_box: minimal length (pix) to keep a box max_candidates: maximum boxes to consider in a single page box_thresh: minimal objectness score to consider a box bin_thresh: threshold used to binzarized p_map at inference time 皙?333333?T box_thresh bin_threshassume_straight_pagesreturnNc\t|||d|_dS)Ng?)super__init__ unclip_ratio)selfr r r __class__s z/var/www/html/Carbon-Document/venv/lib/python3.11/site-packages/doctr/models/detection/differentiable_binarization/base.pyrzDBPostProcessor.__init__ s0 Z1FGGGpointsc|jsotj|}tj|}|dddzd|ddzz}d|dd|ddzzdz}n*tj|}tj|d}||jz|z }tj}| |tj tj | |}d}t|dkrDd} t|D])\} } t| | kr| }t| } *||g}tj|} t| dkrdS|jrtj| n:tjtjtj| ddS) zExpand a polygon (points) by a factor unclip_ratio, and returns a polygon Args: points: The first parameter. Returns: a box in absolute coordinates (xmin, ymin, xmax, ymax) or (4, 2) array (quadrangle) rrT)closedNaxis)r cv2 minAreaRect boxPoints contourArea arcLengthr pyclipperPyclipperOffsetAddPathJT_ROUNDET_CLOSEDPOLYGONExecutelen enumeratenpasarray boundingRectroll) rrrectarealengthdistanceoffset_pointsidxmax_size_idxpexpanded_pointss rpolygon_to_boxzDBPostProcessor.polygon_to_box)s) 8?6**D]4((FGAJNq471:~6D$q'!*tAwqz12Q6FF?6**D]6$777F$++f4*,,vy193MNNN..** w<tj|dddf|dddfz d |kr|j r@|\} } } } | |z | |z | | z|z | | z|z f\}}}}||||||gt+|tjs|jd krt/d |dddfxx|zcc<|ddd fxx|zcc<|t j|t j d |ggl|j sVt3|dkr(t jt j|dd nt jd|jSt3|dkr(t jt j|dd nt jd|jS)a?Compute boxes from a bitmap/pred_map: find connected components then filter boxes Args: pred: Pred map from differentiable binarization output bitmap: Bitmap map computed from pred (binarized) angle_tol: Comparison tolerance of the angle with the median angle across the page ratio_tol: Under this limit aspect ratio, we cannot resolve the direction of the crop Returns: np tensor boxes for the bitmap, each box is a 5-element list containing x, y, w, h, score for the box NrrrT)r Fr)rzEWhen assume straight pages is false a box is a (4, 2) array (polygon)rg)rrdtype)rr?)shaper findContoursastyper*uint8 RETR_EXTERNALCHAIN_APPROX_SIMPLEanymaxminr r,array box_scorer r9squeezelinalgnormappend isinstancendarrayAssertionErrorvstackr(clipr+zerosrA)rr:r;heightwidth min_size_boxboxescontours_contourxywhrscore_boxxminyminxmaxymaxs rbitmap_to_boxeszDBPostProcessor.bitmap_to_boxesVs" RaR(  02&v}}RX'>'>@QSVSjkk !' H' HGvgaaadm''Q'//'!!!Q$-2C2C2C2K2KKlZ[[ ) S -g66 1a%'X1v1q5zAE1q5>TUXYTY[\S].^%_%_tV4PPtWERRt&&) @**622**2:g+>+>??) <47\#9#9T!W|=S=S>TQT T!QQQ$Z 7bAALPP) H! 1a)*UAJQ%RSVWRW[aQa)a&dD$ dD$e<===="$ 33r f8L8L()pqqqQQQT e# QQQT f$  RYbhU|.D.D'EFFGGGG) n7:5zzA~~272:e,,a33328T]eieoKpKpKp p7:5zzA~~272:e,,a33328TZbfblKmKmKm mr)rr T) __name__ __module__ __qualname____doc__floatboolrr*rRr9rh __classcell__)rs@rrrs   &*      $         +  +  + + + + ZBnjBn Bn  BnBnBnBnBnBnBnBnrc|eZdZUdZdZdZdZdZdZe e d<e dd e j d e j d e j d e j d edef dZde j de j de j dee j e j e j ffdZdeeee j fdeeeefdee j e j e j e j ffdZdS)_DBNeta-DBNet as described in `"Real-time Scene Text Detection with Differentiable Binarization" `_. Args: feature extractor: the backbone serving as feature extractor fpn_channels: number of channels each extracted feature maps is mapped to g?r gffffff?r=Tr ư>xsysabepsr ctj||dz tj||dz z}tj||dz tj||dz z}tj|d|dz tj|d|dz z}||z |z dtj||zz|zz }tj|dd}dtj|z } tj| } tj||z| z|z |z} tjtj|||dk| |dk<| S)aCompute the distance for each point of the map (xs, ys) to the (a, b) segment Args: xs : map of x coordinates (height, width) ys : map of y coordinates (height, width) a: first point defining the [ab] segment b: second point defining the [ab] segment eps: epsilon to avoid division by zero Returns: The computed distance rrrg?)r*squaresqrtrU nan_to_numfmin) rsrtrurvrw square_dist_1 square_dist_2 square_distcosin square_sinresults rcompute_distancez_DBNet.compute_distancesR* "qt),,ryad/C/CC  "qt),,ryad/C/CC i!qt ,,ry1!/E/EE },}<RW]]jMjEkEkAknqAqrtS))5))) ]:.. 6CkQTWWXXGBGM=$I$IJJ5ST9Uuqy rpolygoncanvasmaskc|jdks|jddkrtdt|}|jdt j|jdz z|jz }d|D}tj }| |tj tj t j||d}t!j||t jgd|dddf} |dddf} |dddf} |dddf} | | z dz} | | z dz}|dddf| z |dddf<|dddf| z |dddf<t jt jd| dz | d| || f}t jt jd|dz ||d|| f}t j|jd|| f|j }t7|jdD]U}|dz|jdz}|||||||}t j||z dd||<Vt j|d }t)t+d| |jddz }t)t+d| |jddz }t)t+d| |jddz }t)t+d| |jddz }t jd||| z || z |z|| z || z | zfz |||dz||dzf|||dz||dzf<|||fS) aDraw a polygon threshold map on a canvas, as described in the DB paper Args: polygon : array of coord., to draw the boundary of the polygon canvas : threshold map to fill with polygons mask : mask for training on threshold polygons rrz1polygon should be a 2 dimensional array of coordsc,g|]}t|Stuple.0coors r z*_DBNet.draw_thresh_map..s33345;;333rrryN)numr@r)ndimrBAttributeErrorrr/r*power shrink_ratior0r"r#r$r%r&rKr'rfillPolyrDint32rJrI broadcast_tolinspacereshaperVrArangerrUfmax)rrrr polygon_shaper1subjectpaddingpadded_polygonrdrfrergrXrWrsrt distance_mapijabsolute_distance xmin_valid xmax_valid ymin_valid ymax_valids rdraw_thresh_mapz_DBNet.draw_thresh_maps& <1   a 0A 5 5 !TUU U ((  %RXd6G-K-K)KL}Occ337333+--!3Y5OPPP%'Xgooh.G.G.J%K%K TN11"(;;> T@QT^aeTehmTm@mm  : Q. Z!^0KK L L L zJN*Ja,GGH$$rtarget output_shapec P td|Drtdtd|Drtdt|dkr9t t |djn tj }|\}}}t||||f}tj |tj }tj |t} tj |tj } tj |tj } t|D]I\} } t| D] \}}|jddkrd| | |f<|}|jdkr|dddddfxx|zcc<|ddddd fxx|zcc<|}tj|ddd ddf|dddddfz d }tj|d |d fd tj}n|dddd gfxx|zcc<|ddd dgfxx|zcc<|tj}tj|dddd gf|ddddgf|ddd dgf|ddd d gfgd }tj|ddd f|dddfz |dddf|ddd fz }t9|||D] \}}}||jkr,d| | ||d |dd z|d|d d zf<>t=|}|jd tj |j!d z z|j"z }d |D}tGj$}|%|tFj&tFj'|(| }t|dkr-d| | ||d |dd z|d|d d zf<tj)|d*d d }|jdd kst=|j+s-d| | ||d |dd z|d|d d zf<tYj-|| |f|tjgd|.|| | |f| | |f\}| | |f<| | |f< "K| ||j/|j0z z|j0z} ||}| t} | |} | t} || | | fS)Nc3nK|]0}|D]}|jtjkV1dS)N)valuesrAr*float32rtgtts r z&_DBNet.build_target..s@MM MM1qw"*$MMMMMMMrz;the expected dtype of target 'boxes' entry is 'np.float32'.c 3K|]R}|D];}tj|ddddfdk|ddddfdkzVrr)rr*rHrs rrz&_DBNet.build_target..ss__3RUR\R\R^R^__QrvqBQBx!|!!!RaR%1 566_______rzIthe 'boxes' entry of the target is expected to take values between 0 & 1.rr@Fr=rrrrc,g|]}t|Srrrs rrz'_DBNet.build_target..Js<<A&kkAood4q 0 0 2 23344::SUS] ) QF [!Q7 !#,bh!G!G!G !w|4@@@$&H\$L$L$L "$(Q&&aaaAg&&&!+&&&aaaAg&&&!+&&&%E!# !!!Q'0BYqqqRSUVUVUVwEW0W^`!a!aJ " a0@0@)--PQBRBR/SUW X X ^ ^ ` ` g ghjhp q qIIaaa!Qi(((A-(((aaa!Qi(((A-((( ) 1 1 8 8 B BIH%aaa!Qi0%aaa!Qi0%aaa!Qi0%aaa!Qi0  E"$IaaadOi1o,MyYZYZYZ\]Y]ajklklklnokoapOp!q!qJ+.ui+L+L'D#x$"333]biQ#a&1*1Dc!fsSTvXYzFY!YZ &dmmG&|q28DW>W=XZ]^^^X\WkWkmCN;[i=XXXTD-Y7S)^9T9T3;8 t&,,[99T_t=^_bfbqq &&{33 ??4((%,,[99 !((.. 8]K??rN)rr)rirjrkrlrrrrYr rn__annotations__ staticmethodr*rRrmrrrlistdictstrintrrrrrqrqs}LJJL"&4&&&   J J : :     \>@%@% @%j @% rz2:rz1 2 @%@%@%@%DY@T#rz/*+Y@CcM*Y@ rz2:rz2:= > Y@Y@Y@Y@Y@Y@rrq) rnumpyr*r"shapely.geometryrcorer__all__rrqrrrrs $$$$$$))))))  EnEnEnEnEn,EnEnEnPJ@J@J@J@J@J@J@J@J@J@r