igRSrSSKrSSKrSSKJr SSKJr SSKrSr "SS5r g)z This code is refer from: https://github.com/open-mmlab/mmocr/blob/main/mmocr/datasets/pipelines/textdet_targets/fcenet_targets.py N)fft)normcR[U5S:Xde[USUSS-- 5$)Nr:0yE>)lenabs)vecs f/var/www/html/banglarbhumi/venv/lib/python3.13/site-packages/paddleocr/ppocr/data/imaug/fce_targets.py vector_sloper s. s8q== s1vQ$' ((c^\rSrSrSrSU4SjjrSrSrSrSr Sr S r SS jr S r S rS rSrSrSrS\4SjrSrSrSrSrU=r$) FCENetTargetsatGenerate the ground truth targets of FCENet: Fourier Contour Embedding for Arbitrary-Shaped Text Detection. [https://arxiv.org/abs/2104.10442] Args: fourier_degree (int): The maximum Fourier transform degree k. resample_step (float): The step size for resampling the text center line (TCL). It's better not to exceed half of the minimum width. center_region_shrink_ratio (float): The shrink ratio of text center region. level_size_divisors (tuple(int)): The downsample ratio on each level. level_proportion_range (tuple(tuple(int))): The range of text sizes assigned to each level. c >[TU]5 [U[5(de[U[5(de[ U5[ U5:XdeXlX lX0lX@lXPl X`l gN) super__init__ isinstancetupler fourier_degree resample_stepcenter_region_shrink_ratiolevel_size_divisorslevel_proportion_rangeorientation_thr) selfrrrrrrkwargs __class__s r rFCENetTargets.__init__0sv -u55550%8888&'3/E+FFFF,**D'#6 &<#.rc URS:a U[USS9S-RS5- nOU[USS9S-- nURS:a U[USS9S-RS5- nOU[USS9S-- n[R"[R "[R "X4-SS9SS55$)Nraxisrr#rg?)ndimrreshapenparccosclipsum)rvec1vec2 unit_vec1 unit_vec2s r vector_angleFCENetTargets.vector_angleFs 99q=T 3d :CCGLLIT 3d :;I 99q=T 3d :CCGLLIT 3d :;Iyy (=B!GsSTTrcBURS:XdeURSS:deURSS:Xde[U[5(deUS:de[ [ U5S- 5Vs/sHn[ XS-X- 5PM nn[U5n[R"S/U-5nU[U5S-- nSnUS/n [ SU5HnX7-n US-[ U5:a/XUS-:a$US- nUS-[ U5:a XUS-:aM$XU- n U[ U5:a O0XU- n XXS-X- U --n U RU 5 M U RUS5 [R"U 5n U $s snf)zResample n points on a line. Args: line (ndarray): The points composing a line. n (int): The resampled points number. Returns: resampled_line (ndarray): The points composing the resampled line. rrrgrr#) r(shaperintranger rr-r*cumsumfloatappendarray)rlineni length_list total_length length_cumsum delta_lengthcurrent_edge_indresampled_linecurrent_line_lencurrent_edge_end_shiftend_shift_ratio current_points r resample_lineFCENetTargets.resample_lineQsyyA~~zz!}!!!zz!}!!!!S!!!!1u u edge_vec1x adjacent_indadjacent_edge_vectemp_theta_sumtemp_adjacent_thetatheta_sum_scoreadjacent_theta_score poly_center edge_dist dist_scoreposition_scorescore pad_score score_matrixgaussian head_starttail_increment tail_starthead_endtail_endrQrRhorizontal_edge_indsvertical_edge_indsvertical_len_sumhorizontal_len_sums# r rMFCENetTargets.find_head_tails{{a||A!###||A!###/51111 v;?F1I#67J!!"~ 3B7HI!#  )( 3 <=E1q5>J>aCM 1> J$,$:!!#(9(9)(W!X&*&7&7%a(*;A*>'#  0"))*=>!4!hhy1BEE9O#%88,>#?"%%#G ''&q1K Z^k1;Z_{2<I#RVVI%66JXXc(m4N/)D3G,GGE TJ& &E6{Q!#E a! 35:5r"a'" S>) )Eu~6I88SZUa$@AL #e*q.)E#e*q.,AAA773;'#-/&&"((AGs?C881<=>   "&&"22H3u:&!HQ1s5z>A+=?(JSPQQT"' *,)9)9##%|'9'9* &J%~59S[HJ"Q#f+5H"Q#f+5H(")3ZJ %-x(#X.I#X.I>)##;F1Iq 12\q F1I%6VAY23lq F1I%7 *+AA'7$'(!fq!f%5")*AA'7$'(!fq!f%5"#)!,Q/06:LQ:OPQ:R3SS )!,Q/06:LQ:OPQ:R3SS   "&+A.q12Vlength1length2r@resample_point_numresampled_line1resampled_line2s r resample_sidelines FCENetTargets.resample_sidelinesss ~~4144444q!Y__Q%7<1<<<<<q!Q&&&q!Q&&&-////tltrbrblcurrent_center_boxs r generate_center_region_mask)FCENetTargets.generate_center_region_mask;s(E****XXqfbhh7 D!\\"a0N'+'='=n'M $AqH595L5LD$6$66 2 "4DbD!9 %&#.@*AA-BaGK'*-?-BBCcI !'+.@.DDEK !""6:L:L"LMO!"6$:L:L"LMO;/O"Ca"GG)#c+&6&H &8#c*<&=&O&"&8#c*<&=&O&"1c+.23N)!,{1~=5566  A&)!a%0;q1u3EE5566  A&)!a%0;q1u3EE5566  N)!,{1~=5566 &(YYBB/?%@%G%G%Q"#**+=>-4?n 'a@!!rc/n[[U55HUnXnU[U5S- :XaUSnOXS-nURUSUS- S-USUS- S--S-5 MW [U5n[R "U5US-- U-nUR [R5n/n [[U55HcnXn XnU[U5S- :XaUSnOXS-nU S:XaM2Xe- U - n [U 5Hn X[U --n U RU 5 M Me [R "U 5$)zResample one polygon with n points on its boundary. Args: polygon (list[float]): The input polygon. n (int): The number of resampled points. Returns: resampled_polygon (list[float]): The resampled polygon. rrrr^r)r7r r:r-r*r;rr)rpolygonr=lengthr>p1p2r@n_on_each_line new_polygonnumdxdyjpoints r resample_polygonFCENetTargets.resample_polygonsUs7|$ABCL1$$QZU^ MMBqEBqEMa/2a52a5=Q2FF3N O %6{ ((6*lT.ABaG'..rxx8 s7|$A #CBCL1$$QZU^axGs?D3ZAX ""5) % xx $$rcXRSS9- n[R"USS2S45nUSS2S4n[R"U5n[R"XESS5nXVn[R "XSUSU/5nU$)zNormalize one polygon so that its start point is at right most. Args: polygon (list[float]): The origin polygon. Returns: new_polygon (lost[float]): The polygon with start point at right. rr$Nr)rOr*r argsortargminrg) rr temp_polygonrryindex_xindex_yindexrs r normalize_polygonFCENetTargets.normalize_polygons1!55 FF<1% & A **Q-))AbqkN+ nngfowv%GH rcUSS2S4USS2S4S--n[U5[U5- n[R"XB*SUSUS-45nU$)zPerform Fourier transformation to generate Fourier coefficients ck from polygon. Args: polygon (ndarray): An input polygon. fourier_degree (int): The maximum Fourier degree K. Returns: c (ndarray(complex)): Fourier coefficients. Nrry?)rr r*hstack)rrrrPc_fftcs r poly2fourierFCENetTargets.poly2fouriersbAA!33F c&k) IIu_-.6J8J0KL Mrcz[R"XS-5[R"XS- 5:aU$[R"XS-5[R"XS- 5:aUSSS2$[R"XS-5[R"XS- 5:aU$USSS2$)zMake sure the polygon reconstructed from Fourier coefficients c in the clockwise direction. Args: polygon (list[float]): The origin polygon. Returns: new_polygon (lost[float]): The polygon in clockwise point order. rNr#r)r*r )rrrs r clockwiseFCENetTargets.clockwises 66!Q&' (266!Q4F2G+H HH VVAq() *RVVAq6H4I-J JTrT7Nvva*+,rvva8J6K/LL2wrcPURU5nURU5nURX25nURXB5n[R "U5R S5n[R"U5R S5n[R"XV/5nU$)a/Calculate Fourier signature from input polygon. Args: polygon (ndarray): The input polygon. fourier_degree (int): The maximum Fourier degree K. Returns: fourier_signature (ndarray): An array shaped (2k+1, 2) containing real part and image part of 2k+1 Fourier coefficients. r&) rrrrr*realr)imagr)rrrresampled_polygon fourier_coeff real_part image_partfourier_signatures r cal_fourier_signature#FCENetTargets.cal_fourier_signatures!11': 223DE))*;L }E GGM*227; WW]+33G< IIy&=>  rc [U[5(deUup4URn[R"US-S-X44[R S9n[R"US-S-X44[R S9nUGHn[R"X44[R S9n [R"U5RS5n [R"XR[R5S5 URU SU5n [U*US-5Hn U S:waZXX-S4-SU - XlU-SS2SS24--XlU-SS2SS24'XX-S4-SU - X|U-SS2SS24--X|U-SS2SS24'Mc[R"U S:5n [R "[#U 5[R$S9U-nU SS2S4U SS2S4nnXS4U- XnUU4'XS4U- X~UU4'M GM Xg4$)aHGenerate Fourier coefficient maps. Args: img_size (tuple): The image size of (height, width). text_polys (list[list[ndarray]]): The list of text polygons. Returns: fourier_real_map (ndarray): The Fourier coefficient real part maps. fourier_image_map (ndarray): The Fourier coefficient image part maps. rrdtyperr#rrNr^)rrrr*rffloat32rr;r)rrrrrr7argwhereonesr int64)rrrrrkreal_mapimag_maprmaskrrr>yxk_indrrrs r generate_fourier_maps#FCENetTargets.generate_fourier_mapss(E****   88QUQY-RZZ@88QUQY-RZZ@D88QF"((3Dhhtn,,Z8G LL~~bhh7 ; 66wqz1EMA2q1u%6QUAX66t8xAq! '<<=UAq[) QUAX66t8xAq! '<<=UAq[) TCZ0BGGSWRXX>BEad8R1XqA,9Q$,?!,CHAq[),9Q$,?!,CHAq[)& ,!!rc([U[5(deUup4[R"X44[RS9nUHLn[R "U[R S9RS5n[R"XWS5 MN U$)zGenerate text center region mask and geometry attribute maps. Args: img_size (tuple): The image size (height, width). text_polys (list[list[ndarray]]): The list of text polygons. Returns: text_region_mask (ndarray): The text region mask. rrr) rrr*rfrr;rr)rr)rrrrrtext_region_maskrrs r generate_text_region_mask'FCENetTargets.generate_text_region_mask'su(E****88QF"((;Dhht2884<Generate ground truth target on each level. Args: img_size (list[int]): Shape of input image. text_polys (list[list[ndarray]]): A list of ground truth polygons. ignore_polys (list[list[ndarray]]): A list of ignored polygons. Returns: level_maps (list(ndarray)): A list of ground target on each level. rrrrrN)rrr7r r*r;rr)r boundingRectrerbr:rrrrrg)rrr ignore_polysrr lv_size_divslv_proportion_ranger> lv_text_polyslv_ignore_polys level_mapsrrrbox_wbox_h proportionindproportion_range ignore_poly size_divisorcurrent_level_mapslevel_img_size text_region center_regioneffective_maskfourier_real_mapfourier_image_mapss r generate_level_targets$FCENetTargets.generate_level_targetsRs// "99%*3|+<%=>%=%= >',S->'?@'?!2'?@ Dhht2884<O0OP*H ("+r6s,   ) Z Z r