3j4SSKJr SSKJr SSKJr SSKrSSKJr SSSjjr SSjr SSjr Sr "S S \R5r"S S \R5r"S S\R5rSr\SrS SjrS!S"Sjjr"SS\R5r"SS\R5r"SS\R5rS#SjrS$S%SjjrS&S%Sjjrg)') annotations) OrderedDict)LiteralNcUR5nUS:Xa[R"U5nU$US:Xa[R"X!5nU$US:Xa[R"X2S9nU$[ SUSS35e)Nrelu leakyreluprelu)num_parametersinitzactivation layer [s] is not found)lowernnReLU LeakyReLUPReLUNotImplementedError)act_typeinplace neg_slopen_prelulayers e/home/wildlama/miniconda3/lib/python3.13/site-packages/spandrel/architectures/__arch_helpers/block.pyactrs~~H6  L [  Y0 L W @ L"$6xl."QRRcUR5nUS:Xa[R"USS9nU$US:Xa[R"USS9nU$[ SUSS35e) NbatchT)affineinstanceFznormalization layer [r r )rr BatchNorm2dInstanceNorm2dr) norm_typencrs rnormr$ si!IGr$/ L j !!"U3 L"$9)An"UVVrcUR5nUS:XagUS:Xa[R"U5nU$US:Xa[R"U5nU$[ SUSS35e)Nrreflect replicatezpadding layer [r z] is not implemented)rrReflectionPad2dReplicationPad2dr)pad_typepaddingrs rpadr,,ss~~H!|9""7+ L [ ##G, L"OHQ[TU]5 XlgNsuper__init__subself submodule __class__s rr;ConcatBlock.__init__C rcP[R"XRU54SS9nU$)Nr.)dim)torchcatr<r>xoutputs rforwardConcatBlock.forwardGs"Axx{+3 rcfSnURR5RSS5nX-nU$)NzIdentity .. |  |r<__repr__replacer>tmpstrmodstrs rrPConcatBlock.__repr__Ks2"""$,,T59 rr< __name__ __module__ __qualname____firstlineno__r;rJrP__static_attributes__ __classcell__r@s@rr5r5Arr5c4^\rSrSrU4SjrSrSrSrU=r$) ShortcutBlockRc.>[TU]5 Xlgr8r9r=s rr;ShortcutBlock.__init__TrBrc,XRU5-nU$r8rVrGs rrJShortcutBlock.forwardXsXXa[ rcfSnURR5RSS5nX-nU$Nz Identity + |rMrNrOrRs rrPShortcutBlock.__repr__\2!""$,,T59 rrVrWr^s@rraraRr_rrac4^\rSrSrU4SjrSrSrSrU=r$)ShortcutBlockSPSRcc.>[TU]5 Xlgr8r9r=s rr;ShortcutBlockSPSR.__init__erBrcXR4$r8rV)r>rHs rrJShortcutBlockSPSR.forwardis(({rcfSnURR5RSS5nX-nU$rhrOrRs rrPShortcutBlockSPSR.__repr__lrjrrVrWr^s@rrlrlcsrrlc[U5S:Xa([US[5(a [S5eUS$/nUH~n[U[R 5(a*UR 5HnURU5 M ML[U[R5(dMmURU5 M [R "U6$)Nr.rz.sequential does not support OrderedDict input.) len isinstancerrr SequentialchildrenappendModule)argsmodulesmoduler?s r sequentialr~ss 4yA~ d1g{ + +%&VW WAwG fbmm , ,#__. y)/  * * NN6 "  ==' ""rCNANACCNACc [[R"XSSS9[R"XSSS9U(a[U55$S5$)Nr/r.)r1r+r)r~rConv2dr)in_ncout_ncrs rconv_block_2c2rsI  %Q: &a;!H  (, rc U S;d SU SS35e[X$5n U(aUS:wa [X{5OSn US:XaU OSn [R"UUUUU UUUS9n U (a [ U 5OSnU S ;a U(a [ X5OSn[ XX5$U S :Xa0Uc U b [ U S S 9nU(a [ X5OSn[ XX5$[S U 35e)z Conv layer with padding, normalization, activation mode: - ``CNA`` --> Conv -> Norm -> Act - ``NAC`` --> Norm -> Act --> Conv (Identity Mappings in Deep Residual Networks, ECCV16) rzWrong conv mode [r ]zeroNr)r1strider+r2biasgroups)rrrF)rzInvalid conv mode )r3r,rrrr$r~ ValueError)rrr1rr2rrr*r"rmoder+pcans r conv_blockrs( ) )H->tAha+HH ) 6G"*x6/AHtA!V+gG    A"H tA '0D #d!%%   !5He,A '0D "T!%%-dV455rcP^\rSrSrSrSSU4SjjjrSrSrU=r$) ResNetBlockz ResNet Block, 3-3 style with extra residual scaling used in EDSR (Enhanced Deep Residual Networks for Single Image Super-Resolution, CVPRW 17) c >[TU]5 [UUUUUUUU U U U 5 nU S:XaSn U S:XaSn Sn [UUUUUUUU U U U 5 n[X5UlXlg)Nrr)r:r;rr~res res_scale)r>rmid_ncrr1rr2rrr*r"rrrconv0conv1r@s rr;ResNetBlock.__init__s               5=H 6>HI             &e+"rc^URU5RUR5nX-$r8)rmulr)r>rHrs rrJResNetBlock.forward s#hhqkoodnn-wr)rr) r.r.r.TrNrrr.)rConvMode rXrYrZr[__doc__r;rJr\r]r^s@rrrsG 7#7#7#rrrcV^\rSrSrSrSSU4SjjjrSrSrU=r$)RRDBizf Residual in Residual Dense Block (ESRGAN: Enhanced Super-Resolution Generative Adversarial Networks) c >[T U]5 [UUUUUUUUU U S9 Ul[UUUUUUUUU U S9 Ul[UUUUUUUUU U S9 Ulg)N)plus)r:r;ResidualDenseBlock_5CRDB1RDB2RDB3)r>nfr1gcrrr*r"rr _convtype_spectral_normrr@s rr; RRDB.__init__s )            *            *            rcxURU5nURU5nURU5nUS-U-$)N皙?rrr)r>rHouts rrJ RRDB.forwardLs7iiliiniinSy1}rr) r r.TrNrrConv2DFFrboolrrrr^s@rrrsT  3  3 3 3 jrrcT^\rSrSrSrSSU4SjjjrSrSrU=r$)riSa Residual Dense Block Style: 5 convs The core module of paper: (Residual Dense Network for Image Super-Resolution, CVPR 18) Modified options that can be used: - "Partial Convolution based Padding" `arXiv:1811.11718` - "Spectral normalization" `arXiv:1802.05957` - "ICASSP 2020 - ESRGAN+ : Further Improving ESRGAN" N. C. {Rakotonirina} and A. {Rasoanaivo} Args: nf (int): Channel number of intermediate features (num_feat). gc (int): Channels for each growth (num_grow_ch: growth channel, i.e. intermediate channels). convtype (str): the type of convolution to use. Default: 'Conv2D' gaussian_noise (bool): enable the ESRGAN+ gaussian noise (no new trainable parameters) plus (bool): enable the additional residual paths from ESRGAN+ (adds trainable parameters) c n>[T U]5 U (a [X5OSUl[UUUUUUUUU S9 Ul[X-UUUUUUUU S9 Ul[USU--UUUUUUUU S9 Ul[USU--UUUUUUUU S9 UlU S:XaSn OUn [USU--USUUUUU U S9 Ulg)N)rr*r"rrr/rr) r:r;conv1x1rrconv2conv3conv4conv5) r>rr1rrrr*r"rrrlast_actr@s rr;ResidualDenseBlock_5C.__init__fs +/wrD         G       RK       RK       5=HH RK     rc URU5nUR[R"X4S55nUR(aX0R U5-nUR [R"XU4S55nUR [R"XX44S55nUR(aXS-nUR[R"XX4U4S55nUS-U-$)Nr.r)rrrErFrrrr)r>rHx1x2x3x4x5s rrJResidualDenseBlock_5C.forwards ZZ] ZZ 1'1- . <<ll1o%B ZZ 1"+q1 2 ZZ 1"/15 6 <<B ZZ 1"""5q9 :Cx!|r)rrrrrr) @rrr.TrNrrFrrr^s@rrrSsU(   M  M M M ^  rrc0[R"XSUSS9$)Nr.F)r1rr)rr) in_planes out_planesrs rrrs 99Y&u UUrc [UXS--UUUUSSS9n [R"U5n U(a [Xq5OSn U(a [ U5OSn [ XX5$)z Pixel shuffle layer (Real-Time Single Image and Video Super-Resolution Using an Efficient Sub-Pixel Convolutional Neural Network, CVPR17) r/Nrr*r"r)rr PixelShuffler$rr~) rrupscale_factorr1rrr*r"rconv pixel_shufflerrs rpixelshuffle_blockrsf  !#$  DOON3M#,Y$A!H tA d1 00rc `[R"X)S9n [UUUUUUUUS9n [X5$)N) scale_factorrr)rUpsamplerr~) rrrr1rrr*r"rrupsamplers r upconv_blockrsA{{BH    D h %%r)Trr.)rstr)r"rr#int)r*r)r)r.r.r.TrNrr) rrrrr" str | Nonerrrr)r.)r/rr.TrNr)rrrrr"r)r/rr.TrNrnearest) __future__r collectionsrtypingrrEtorch.nnrrr$r,r3rzr5rarlr~rrrrrrrrrr0rrrsq##     "))"BII"   # ' (    !16 16 161616 16rB"))BJ?299?DmBIIm`V  1 1 11H   & & &&r