9i SSKJr "SS\R5rS SjrSr"SS\R5r"SS \R5rg) Nc6^\rSrSrSrSU4SjjrSrSrU=r$)FPNFusionModulez7This is a fpn-style cross-scale feature fusion module" cf>[TU]5 XlX lX0l[ XXCS9Ulg)N)use_bnn_block)super__init__ embed_dimsfuse_dimr_make_multi_scale_layers multi_scaler)selfr r rr __class__s d/var/www/html/land-doc-ocr/venv/lib/python3.13/site-packages/modelscope/models/cv/vidt/fpn_fusion.pyr FPNFusionModule.__init__ s9  %  5 BcUn[URS- SS5HJn[URSUS35"X5X'[URSU35"X5X'ML /n[URS- SS5HnX RS- :Xa$[URSU35"X/S5nO7[URSU35"WX/XR 5SS5nUR U5 M U$)Nlayer__rnp_norm_ refinenet_)rangergetattrrsizeappend)rx_blocksidxrefined_embedspaths rforwardFPNFusionModule.forwards()2r2C#D$5$5uC7HI HM#D$5$5G HM3)2r2CllQ&&t00!+C5134[[U] 5 XPlSUlX@lUnUR SLaUS-n[ R"UUSSSS9Ul[XU5Ul [XU5Ul [ RR5Ulg)zInit. Args: features (int): channel dim of the input feature activation: activation function to use bn: whether to use bn expand: whether to expand feature or not align_corners: whether to use align_corners for interpolation rTr)r2r?rBN)r rJr rIrBrHr5r8 smoothingResidualConvUnit resConfUnit1 resConfUnit2 quantizedFloatFunctionalskip_add)rrL activationrGrHrI out_featuresrs rr FeatureFusionBlock.__init__gs  $02*   ;;$ #q=L    -X2F,X2F 446 rc$USn[U5S:Xaj[RRUUSURS9nUR U5nUR US5nURRX45nURU5nU$)a!Forward pass. Args xs: xs[0]: the feature refined from the previous step, xs[1]: the next scale features to fuse up_size: the size for upsampling; xs[0] is upsampled before merging with xs[1] Returns: output: the fused feature, which is fed to the next fusion step as an input rrbilinear)rmoderIr) lenr5 functional interpolaterIrPrRrVaddrS)rxsup_sizeoutputress rr$FeatureFusionBlock.forwardsA r7a<]].."00 /2F ^^F+F##BqE*C]]&&v3F""6* r)rIrHrBrRrSrVrP)FFTr(r0s@rrJrJds! # "7HrrJc2^\rSrSrSrU4SjrSrSrU=r$)rQzResidual convolution module. c >[TU]5 X0lSUl[R "USSSUR(+URS9Ul[R "SSSSSUR(+URS9Ul[R "SUSSUR(+URS9UlURSLaQ[R"U5Ul [R"U5Ul [R"U5Ul X l [RR5Ulg)zwInit. Args: features (int): channel dim of the input activation: activation function bn: whether to use bn r@)r2strider?rB)r2rjpaddingr?rBTN)r r rGrBr5r8conv1conv2conv3 BatchNorm2dbn1bn2bn3rWrTrUrV)rrLrWrGrs rr ResidualConvUnit.__init__s  YY  WW;;   YY  WW;;  YY  WW;;    77d?~~h/DH~~h/DH~~h/DH$ 446 rcURU5nURU5nURSLaURU5nURU5nUR U5nURSLaUR U5nURU5nUR U5nURSLaURU5nURS:aURU5nURRX!5$)zWForward pass Args: x (tensor): input feature Returns: tensor: output feature Tr) rWrmrGrqrnrrrorsrB conv_mergerVr`)rxouts rr$ResidualConvUnit.forwardsooa jjo 77d?((3-Cooc"jjo 77d?((3-Cooc"jjo 77d?((3-C ;;?//#&C}}  ((r) rWrGrqrrrsrmrnrorBrVr(r0s@rrQrQs(,7\))rrQ)r'rF)torch.nnr5r6rr r:rJrQrrr|sV ,bii,b&'$%$) B DDNM)ryyM)r