o ߥiQ@snddlZddlmZddlmZddlmZmZmZGdddej Z Gdd d ej Z Gd d d ej Z dS) N)nn) functional)default_init_weights make_layerpixel_unshufflecs*eZdZdZdfdd ZddZZS) ResidualDenseBlockzResidual Dense Block. Used in RRDB block in ESRGAN. Args: num_feat (int): Channel number of intermediate features. num_grow_ch (int): Channels for each growth. @ cstt|t||ddd|_t|||ddd|_t|d||ddd|_t|d||ddd|_t|d||ddd|_ tj ddd|_ t |j|j|j|j|j gddS) Nr皙?Tnegative_slopeinplaceg?) superr__init__rConv2dconv1conv2conv3conv4conv5 LeakyReLUlrelurselfnum_feat num_grow_ch __class__f/home/ubuntu/.local/lib/python3.10/site-packages/modelscope/models/cv/super_resolution/rrdbnet_arch.pyrszResidualDenseBlock.__init__c Cs|||}||t||fd}||t|||fd}||t||||fd}|t|||||fd}|d|S)Nrr)rrrtorchcatrrr)rxx1x2x3x4x5r"r"r#forward$s   zResidualDenseBlock.forward)r r __name__ __module__ __qualname____doc__rr, __classcell__r"r"r r#r s rcs*eZdZdZdfdd ZddZZS)RRDBzResidual in Residual Dense Block. Used in RRDB-Net in ESRGAN. Args: num_feat (int): Channel number of intermediate features. num_grow_ch (int): Channels for each growth. r cs6tt|t|||_t|||_t|||_dS)N)rr3rrrdb1rdb2rdb3rr r"r#r8s  z RRDB.__init__cCs*||}||}||}|d|S)Nr)r4r5r6)rr&outr"r"r#r,>s    z RRDB.forward)r r-r"r"r r#r3.s r3cs2eZdZdZ    d fdd Zdd ZZS) RRDBNeta+Networks consisting of Residual in Residual Dense Block, which is used in ESRGAN. ESRGAN: Enhanced Super-Resolution Generative Adversarial Networks. We extend ESRGAN for scale x2 and scale x1. Note: This is one option for scale 1, scale 2 in RRDBNet. We first employ the pixel-unshuffle (an inverse operation of pixelshuffle to reduce the spatial size and enlarge the channel size before feeding inputs into the main ESRGAN architecture. Args: num_in_ch (int): Channel number of inputs. num_out_ch (int): Channel number of outputs. num_feat (int): Channel number of intermediate features. Default: 64 num_block (int): Block number in the trunk network. Defaults: 23 num_grow_ch (int): Channels for each growth. Default: 32. r r r cstt|||_|dkr|d}n|dkr|d}t||ddd|_tt|||d|_ t||ddd|_ t||ddd|_ t||ddd|_ t||ddd|_ t||ddd|_tjddd |_dS) Nr r rr )rrrTr)rr8rscalerr conv_firstrr3body conv_bodyconv_up1conv_up2conv_hr conv_lastrr)r num_in_ch num_out_chr;r num_blockrr r"r#rZs  zRRDBNet.__init__c Cs|jdkr t|dd}n|jdkrt|dd}n|}||}|||}||}||tj|ddd}|| tj|ddd}| || |}|S)Nr )r;rr nearest) scale_factormode) r;rr<r>r=rr?F interpolater@rBrA)rr&feat body_featr7r"r"r#r,ss   zRRDBNet.forward)r r r9r r-r"r"r r#r8Fsr8) r$rtorch.nnrrI arch_utilrrrModulerr3r8r"r"r"r#s  $