i!!bSrSSKrSSKrSSKrSSKJr SSKJrJ r Sr S Sjr "SS\ 5r g) zv This code is refer from: https://github.com/open-mmlab/mmocr/blob/v0.3.0/mmocr/models/textdet/postprocess/wrapper.py N)ifft)poly_nmsvalid_boundarycURup[R"US-US-4[R5nUR 5USUS-2SUS-24'[R"US-US-4[R5n[ R "X4SS5 USUS-2SUS-24R[R5nU)U-$)N)rr) shapenpzerosuint8copycv2 floodFillastypebool_) input_maskhwcanvasmasks k/var/www/html/banglarbhumi/venv/lib/python3.13/site-packages/paddleocr/ppocr/postprocess/fce_postprocess.py fill_holers   DA XXq1ua!enbhh /F#-??#4F1q1u9a!a%i  88QUAENBHH -DMM&* AAIq1q5y( ) 0 0 :F 7Z c[R"[U5U4SS9n[US5S- S-nUSS2US24USS2SUS-24'USS2SU24USS2U*S24'[U5U-n[R"[U5US45nURUSS2SS2S4'UR USS2SS2S4'UR S5R[U5S45$) aLInverse Fourier transform Args: fourier_coeff (ndarray): Fourier coefficients shaped (n, 2k+1), with n and k being candidates number and Fourier degree respectively. num_reconstr_points (int): Number of reconstructed polygon points. Returns: Polygons (ndarray): The reconstructed polygons shaped (n, n') complexdtyperrrNint32)r r lenrrealimagrreshape) fourier_coeffnum_reconstr_pointsak poly_complexpolygons r fourier2polyr+'s #m$&9:)LA ]1   "q(A#AqrE*AaQUlOa!e$Aa!fI700LhhM*,?CDG#((GAq!G#((GAq!G >>' " * *C ,>+C DDrc^\rSrSrSrS SjrSrSrSrSr S Sjr S r g ) FCEPostProcess?z The post process for FCENet. c  pXlX lX0lX@lXPlX`lXplXlXlg)N) scalesfourier_degreer& decoding_type score_thrnms_thralphabetabox_type) selfr0r1r&r2r3r4r5r6r7kwargss r__init__FCEPostProcess.__init__Ds4 ,#6 *"    rc /nUR5HhupE[U[R5(aUR 5nUSS2SS2SS2SS24nUSS2SS2SS2SS24nUR Xg/5 Mj UR X25$)Nr )items isinstancepaddleTensornumpyappend get_boundary)r8preds shape_list score_mapskeyvaluecls_resreg_ress r__call__FCEPostProcess.__call__[s ++-JC%// Arr1aK(GAqr1aK(G   w0 1 (  88rc /n/nUHn[U5n[US5 URUS5 [R"USUS- 5[R "USS[ US- S- 55RSUS- 5-R5R5nUR[R"U5RSS/55 M [R"U[RS9U4$)aRescale boundaries via scale_factor. Args: boundaries (list[list[float]]): The boundary list. Each boundary with size 2k+1 with k>=4. scale_factor(ndarray): The scale factor of size (4,). Returns: boundaries (list[list[float]]): The scaled boundaries. Tr Nrrr) r!rrBr arraytileintr$flattentolistfloat32)r8 boundaries scale_factorboxesscoresbszs rresize_boundaryFCEPostProcess.resize_boundaryfsAQB 1d # MM!B% HHQxa[)ww|BQ/b1f\1BCKKArTUvVX  LL!,,b!W5 6xxRZZ0&88rc`[U5[UR5:Xde/n[U5H'upEURUnX0RXV5-nM) [ X0R 5nUR USUSSS24- R5SSS25up7[X7S9/nU$)Nrrrr )pointsrW) r!r0 enumerate_get_boundary_singlerr4rZrRdict) r8rFrErTidx score_mapscalerW boxes_batchs rrCFCEPostProcess.get_boundarys:#dkk"2222 ' 3NCKK$E#&?&? &QQJ4 j,,7 !11 Z12..6682>  :=> rc .[U5S:XdeUSRSSUR-S-:XdeURUURURUUR UR URURURS9 $)Nrrr ) rDr1r&rcr5r6r7r3r4) r!r r1 fcenet_decoder&r5r6r7r3r4)r8rbrcs rr_#FCEPostProcess._get_boundary_singles9~"""|!!!$D,?,?(?!(CCCC!!.. $ 8 8**]]nnLL"  rc [U[5(de[U5S:XdeUS;deUSSn U SSn U SSn USSR/SQ5n U SS2SS2SSU-S-24nU SS2SS2SU-S-S24nU SU-U SU--nUU:n[ U5n[ R "UR[R5[ R[ R5unn[R"U5n/nUGHnUR5R[R5n[ R"UU/SSS5 UU-nUS:n[R "U5nUSS2S4USS2S4S--nUUUUnnUUS--nUSS2U4U-USS2U4'UU-n[#UU5n UUR%SS5n!['[R("U U!45R+5U 5n UU -nGM ['UU 5nUS :Xa/n"UHn#[R,"U#SS5R%SS5R[R.5n$U#Sn![ R0"[ R2"U$55n%[R4"U%5n%U"R7U%R%S5R+5U!/-5 U"nM U$) adDecoding predictions of FCENet to instances. Args: preds (list(Tensor)): The head output tensors. fourier_degree (int): The maximum Fourier transform degree k. num_reconstr_points (int): The points number of the polygon reconstructed from predicted Fourier coefficients. scale (int): The down-sample scale of the prediction. alpha (float) : The parameter to calculate final scores. Score_{final} = (Score_{text region} ^ alpha) * (Score_{text center region}^ beta) beta (float) : The parameter to calculate final score. box_type (str): Boundary encoding type 'poly' or 'quad'. score_thr (float) : The threshold used to filter out the final candidates. nms_thr (float) : The threshold of nms. Returns: boundaries (list[list[float]]): The instance boundary and confidence list. r)polyquadrNr)rrrr y?rk)r>listr! transposerr findContoursrr r RETR_TREECHAIN_APPROX_SIMPLE zeros_likerint8 drawContoursargwherer+r$rhstackrRrNrS boxPoints minAreaRectint64rB)&r8rDr1r&rcr5r6r7r3r4cls_predtr_predtcl_predreg_predx_predy_pred score_pred tr_pred_masktr_mask tr_contours_rrTcontdeal_maprb score_maskxy_textdxyxycpolygonsscorenew_boundariesboundaryrjr]s& rrgFCEPostProcess.fcenet_decodesB%&&&&5zQ++++8A;1Q-AB<8A;((3!Q 8!n"4q"8 889!QN 2Q 6 889ajE)hqkT.AB "i/ L))) NN288 $cmmS5L5L  Q}}W% Dyy{))"''2H   Xvr1b 9"X-I"QJkk*-G!Q$-'!Q$-""44C*%vj'9qAAF A#$Q%6#7##=Aa JA#A':;Hj)11"a8E 8U*; < C C EwOH#h.J% (j'2 v N&xx" .66r1=DDRZZP  st'<=&)%%fnnR&8&?&?&AUG&KL+ 'r) r5r6r7r2r1r4r&r0r3N)2fcenet333333?皙??rrj)rg@rjrr) __name__ __module__ __qualname____firstlineno____doc__r:rKrZrCr_rg__static_attributes__rrr-r-?sQ !. 99:  , Wrr-)r)rrr?rAr numpy.fftrppocr.utils.poly_nmsrrrr+objectr-rrrrs4  9  E0{V{r