Apppublic
ReflectionEraser/ReflectionEraserApp

sourceHugging Faceotherupdated 2y agoView on Hugging Face
0likes
rootDSRNetmodelsarch__pycache____init__.cpython-38.pyc
__init__.cpython-38.pyc78 linesDownload Raw Back to __pycache__
1U

2�wdR��@s8ddlZddlmZmZmZmZmZmZmZddl	m3Z4ddlmZm
Z
mZmZddlmZddlmZddlmZddlmZdd	lmZmZmZdd5lmZmZddlm Z ddl!m"Z"dd
l#m$Z$ddl%m&Z&m'Z'ddl(m)Z)ddl*m+Z+ddl,m-Z-ddl.m/Z/ddl0m1Z1m2Z2m3Z3m4Z4m5Z5m6Z6m7Z7ddl8m9Z9m:Z:m;Z;ddl<m=Z=ddl>m?Z?ddl@mAZAddlBmCZCddlDmEZEddlFmGZGddlHmIZIddlJmKZKmLZLddlMTddlNTddlOTddlPTddlQTddlRmSZSmTZTddlUmVZVd d!�ZWd"d#�ZXd$d%�ZYd&d'�ZZd(d)�Z[d�d,d-�Z\d�d.d/�Z]d�d0d1�Z^d�d2d3�Z_d�d4d5�Z`d�d7d8�Zad�d9d:�Zbd�d;d<�Zcd�d=d>�Zdd�d?d@�Zed�dAdB�Zfd�dDdE�Zgd�dFdG�Zhd�dHdI�Zid�dJdK�Zjd�dLdM�Zkd�dNdO�Zld�dPdQ�Zmd�dRdS�Znd�dTdU�Zod�dVdW�Zpd�dXdY�Zqd�dZd[�Zrd�d\d]�Zsd�d^d_�Ztd�d`da�Zud�dbdc�Zvd�ddde�Zwd�dgdh�Zxd�didj�Zyd�dkdl�Zzd�dmdn�Z{d�dodp�Z|d�dqdr�Z}d�dsdt�Z~d�dudv�Zd�dwdx�Z�d�dydz�Z�d�d{d|�Z�d�d}d~�Z�d�dd��Z�d�d�d��Z�d�d�d��Z�d�d�d��Z�d�d�d��Z�d�d�d��Z�d�d�d��Z�d�d�d��Z�d�d�d��Z�d�d�d��Z�d�d�d��Z�d�d�d��Z�d�d�d��Z�d�d�d��Z�d�d�d��Z�d�d�d��Z�d�d�d��Z�d�d�d��Z�d�d�d��Z�d�d�d��Z�d�d�d��Z�d�d�d��Z�d�d�d��Z�d�d�d��Z�e�d�k�r4dd�l�m�Z�ddl�Z�e���d�d*d�d�����Z�exd*d*����Z�eVe��eqd*d*����Z�eVe��e���d�����Z�e����Z�e�d��D]Z�e�e���qe����Z�e�e�e�d��dS)��N)�FINet_Base_R2C_Ablation1�FINet_Base_R2C_Ablation2�FINet_Base_R2C_Ablation4�FINet_Base_R2C_Ablation5�FINet_Base_R2C_Ablation7�FINet_Base_R2C_Ablation9�FINet_Base_R2C_Ablation14��6FINet_Base)�
FINet_Base_R2�FINet_Base_R2B�FINet_Base_R2C�FINet_Base_ResNet_R2C��FINet_LightSpy_BR2C��FINet_LightSpy_BR2C_Dec��FINet_LightSpy_BR2C_Dec2��FINet_LightSpy_CR2C)�FINet_LightSpy_R2C�FINet_LightSpy_Small_R2C�FINet_LightSpy_Tiny_R2C)�FINet_MuGI_R2C�FINet_MuGI_R2C2��FINet��FINet_Light��FINetV2)�	FINetV2_2�
FINetV2_2_R2C��	FINetV2_3��	FINetV2_4��	FINetV2_6��FINetV3)�YTMT_NAFNet2�YTMT_NAFNet3�YTMT_NAFNet3RepVGG�YTMT_NAFNet3RepVGGB3�YTMT_NAFNet3RepVGGPlus�YTMT_NAFNet3RepVGGPlus48�YTMT_NAFNet3RepVGGB3_MaskEnc)�YTMT_NAFNet3SFPRepVGGB3�YTMT_NAFNet4�YTMT_NAFNet3TFPRepVGGB3��YTMT_NAFNet3RepVGGB3_Add��YTMT_NAFNet3RepVGGB3_Concat��#YTMT_NAFNet3RepVGGB3WithMask_Concat��YTMT_NAFNet3RepVGGB3Fex��#YTMT_NAFNet3RepVGGB3_Concat_MaskEnc��,YTMT_NAFNet3RepVGGB3_Concat_MaskEnc_MaskGate��%YTMT_NAFNet3RepVGGB3_MaskEnc_MaskGate)�)YTMT_NAFNet3RepVGGB3_Mix_MaskEnc_MaskGate�*YTMT_NAFNet3RepVGGB3_Mix2_MaskEnc_MaskGate)�*)�
YTMTRestormer�YTMTRestormer2)�
print_networkcKst||fddi|��S�N�pattern�A��YTMT_PS��in_channels�out_channels�kwargs�rS�>D:\workspace\python\YTMT-Strategy-main\models\arch\__init__.py�ytmt_pas*srUcKst||fddi|��S�NrK�CrMrOrSrSrT�ytmt_pcs.srXcKst||fddi|��SrJ��YTMT_USrOrSrSrT�ytmt_uas2sr[cKst||f|�S�N)�YTMTUNetrOrSrSrT�ytmt_ucs_old6sr^cKst||fddi|��SrVrYrOrSrSrT�ytmt_ucs:sr_�� cCs0ddddg}d}ddddg}t||||||d�S�N������width�middle_blk_num�enc_blk_nums�dec_blk_nums)�YTMT_NAFNet�rPrQrh�enc_blksri�dec_blksrSrSrT�
ytmt_nafnet_m>s�rpcCs0ddddg}d}ddddg}t||||||d�Srb�r,rmrSrSrT�ytmt_nafnet_m2Is�rrcCs0ddddg}d}ddddg}t||||||d�SrbrqrmrSrSrT�ytmt_nafnet_m2_repvggTs�rscCs0ddddg}d}ddddg}t||||||d�Srb)r-rmrSrSrT�ytmt_nafnet_m3_s�rtcCs0ddddg}d}ddddg}t||||||d�Srb)r.rmrSrSrT�ytmt_nafnet_m3_repvggjs�ru�0cCs0ddddg}d}ddddg}t||||||d�Srb�r/rmrSrSrT�ytmt_nafnet_m3_repvggb3us�rxcCs0ddddg}d}ddddg}t||||||d�Srbr6rmrSrSrT�ytmt_nafnet_m3_repvggb3_add�s�rycCs0ddddg}d}ddddg}t||||||d�Srbr<rmrSrSrT�ytmt_nafnet_m3_repvggb3_fex�s�rzcCs0ddddg}d}ddddg}t||||||d�Srbr8rmrSrSrT�ytmt_nafnet_m3_repvggb3_concat�s�r{cCs0ddddg}d}ddddg}t||||||d�S�Nrcrg)r3rmrSrSrT�ytmt_nafnet_tiny_repvggb3�s�r}cCs0ddddg}d}ddddg}t||||||d�S�Nrc�rg)r5rmrSrSrT�ytmt_nafnet_tiny2_repvggb3�s�r��cCs0ddddg}d}ddddg}t||||||d�Sr~rrmrSrSrT�finet�s�r�cCs0ddddg}d}ddddg}t||||||d�Srbr	rmrSrSrT�7finet_base�s�r�cCs0ddddg}d}ddddg}t||||||d�Srb)rrmrSrSrT�
finet_base_r2�s�r�cCs0ddddg}d}ddddg}t||||||d�Srb)rrmrSrSrT�finet_base_r2c_ablation1�s�r�cCs0ddddg}d}ddddg}t||||||d�Srb)rrmrSrSrT�finet_base_r2c_ablation2�s�r�cCs0ddddg}d}ddddg}t||||||d�Srb)rrmrSrSrT�finet_base_r2c_ablation4�s�r�cCs0ddddg}d}ddddg}t||||||d�Srb)rrmrSrSrT�finet_base_r2c_ablation5�s�r�cCs0ddddg}d}ddddg}t||||||d�Srb)rrmrSrSrT�finet_base_r2c_ablation7s�r�cCs0ddddg}d}ddddg}t||||||d�Srb)rrmrSrSrT�finet_base_r2c_ablation9s�r�cCs0ddddg}d}ddddg}t||||||d�Srb�rrmrSrSrT�finet_base_r2c_ablation14s�r�cCs0ddddg}d}ddddg}t||||||d�S�Nrcrdrergr�rmrSrSrT�finet_base_r2c_ablation14_2%s�r�cCs0ddddg}d}ddddg}t||||||d�Srb)rrmrSrSrT�finet_base_r2b0s�r�cCs0ddddg}d}ddddg}t||||||d�Srb)r
rmrSrSrT�finet_base_r2c;s�r�cCs0ddddg}d}ddddg}t||||||d�Srb)rrmrSrSrT�finet_mugi_r2cFs�r�cCs0ddddg}d}ddddg}t||||||d�Srb)rrmrSrSrT�finet_mugi_r2c2Qs�r�cCs0ddddg}d}ddddg}t||||||d�Srb)rrmrSrSrT�finet_base_resnet_r2c\s�r�cCs0ddddg}d}ddddg}t||||||d�Srb)�FINet_Base_R2DrmrSrSrT�finet_base_r2dgs�r��@cCs0ddddg}d}ddddg}t||||||d�Srb)rrmrSrSrT�finet_light_spy_r2crs�r�cCs0ddddg}d}ddddg}t||||||d�SrbrrmrSrSrT�finet_light_spy_cr2c}s�r�cCs0ddddg}d}ddddg}t||||||d�SrbrrmrSrSrT�finet_light_spy_br2c�s�r�cCs0ddddg}d}ddddg}t||||||d�SrbrrmrSrSrT�finet_light_spy_br2c_dec�s�r�cCs0ddddg}d}ddddg}t||||||d�SrbrrmrSrSrT�finet_light_spy_br2c_dec2�s�r�cCs0ddddg}d}ddddg}t||||||d�Sr�)rrmrSrSrT�finet_light_small_spy_r2c�s�r�cCs0ddddg}d}ddddg}t||||||d�Sr�)rrmrSrSrT�finet_light_tiny_spy_r2c�s�r�cCs0ddddg}d}ddddg}t||||||d�Sr|r rmrSrSrT�
finetv2_small�s�r�cCs0ddddg}d}ddddg}t||||||d�Sr��r"rmrSrSrT�finetv2_2_b1�s�r�cCs0ddddg}d}ddddg}t||||||d�Srbr�rmrSrSrT�finetv2_2_b2�s�r�cCs0ddddg}d}ddddg}t||||||d�Srb)r#rmrSrSrT�finetv2_2_b2_r2c�s�r�cCs0ddddg}d}ddddg}t||||||d�S)Nrd�rerfrcrgr�rmrSrSrT�finetv2_2_b3�s�r�cCs0ddddg}d}ddddg}t||||||d�Srbr(rmrSrSrT�finetv2_6_b3�s�r�cCs0ddddg}d}ddddg}t||||||d�S)Nrcrdrgr*rmrSrSrT�8finetv3_b1s�r�cCs4dddddg}d}dddddg}t||||||d�Sr|r$rmrSrSrT�finetv2_3_b1s�r�cCs0ddddg}d}ddddg}t||||||d�Srbr&rmrSrSrT�finetv2_4_b1s�r�rfcCs0ddddg}d}ddddg}t||||||d�Sr~rrmrSrSrT�finet_light"s�r�cCs0ddddg}d}ddddg}t||||||d�Sr�rrmrSrSrT�finet_light_b2-s�r�cCs0ddddg}d}ddddg}t||||||d�S�Nrcrdr�rgrrmrSrSrT�finet_light_b38s�r�cCs9t||�Sr\)r4)rPrQrSrSrT�ytmt_nafnet4_repvggb3Csr�cCs0ddddg}d}ddddg}t||||||d�Sr�rwrmrSrSrT�ytmt_nafnet_s3_repvggb3Gs�r�cCs0ddddg}d}ddddg}t||||||d�Sr�)r2rmrSrSrT�ytmt_nafnet_s3_repvggb3_maskencRs�r��$c	Cs2ddddg}d}ddddg}t|||||||d�S�Nrcrdr���
content_width�10mask_widthrirjrkr>�rPrQr�r�rnrirorSrSrT�&ytmt_nafnet_s3_repvggb3_concat_maskenc]s�r�c	Cs2ddddg}d}ddddg}t|||||||d�Sr�rBr�rSrSrT�(ytmt_nafnet_s3_repvggb3_maskenc_maskgatejs�r�c	Cs2ddddg}d}ddddg}t|||||||d�Sr�)rDr�rSrSrT�,ytmt_nafnet_s3_repvggb3_mix_maskenc_maskgatews�r�c	Cs2ddddg}d}ddddg}t|||||||d�Sr��rEr�rSrSrT�-ytmt_nafnet_s3_repvggb3_mix2_maskenc_maskgate�s�r��(rec	Cs2ddddg}d}ddddg}t|||||||d�S)Nrcrdrerfr�r�r�rSrSrT�-ytmt_nafnet_m3_repvggb3_mix2_maskenc_maskgate�s�r�c	Cs2ddddg}d}ddddg}t|||||||d�Sr�r@r�rSrSrT�/ytmt_nafnet_s3_repvggb3_concat_maskenc_maskgate�s�r�cCs0ddddg}d}ddddg}t||||||d�Sr�r:rmrSrSrT� ytmt_nafnet_s3_repvggb3_maskgate�s�r�cCs0ddddg}d}ddddg}t||||||d�S)Nrcrdr�re�rg)r0rmrSrSrT�ytmt_nafnet_m3_repvggplus�s�r�cCs0ddddg}d}ddddg}t||||||d�Srb)r1rmrSrSrT�ytmt_nafnet_m3_repvggplus48�s�r�cCs0ddddg}d}ddddg}t||||||d�Srb)�YTMT_NAFNet3RepVGGReLUrmrSrSrT�ytmt_nafnet_m3_repvgg_relu�s�r�cCs,d}ddddg}d}d}t||||||d�S�Nrarcrdr��@��dim�11num_blocks�num_refinement_blocks�ffn_expansion_factor�rG�rPrQr�r�r�r�rSrSrT�ytmt_restormer_s�s�r�cCs,d}ddddg}d}d}t||||||d�Sr��rHr�rSrSrT�ytmt_restormer2_s�s�r�cCs,d}ddddg}d}d}t||||||d�S)Nrvrcrdr�r�r�r�r�rSrSrT�ytmt_restormer2_m�s�r�cCs,d}ddddg}d}d}t||||||d�S)Nrvrcrdr�g�������?r�r�r�rSrSrT�ytmt_restormer_m�s�r��__main__)�mutilsr�)rr`r�r��2)r`r`ra)r`r`ra)r`r`ra)r`r`ra)r`r`ra)r`r`rv)r`r`rv)r`r`rv)r`r`rv)r`r`ra)r`r`ra)r`r`r�)r`r`rv)r`r`rv)r`r`rv)r`r`rv)r`r`rv)r`r`rv)r`r`rv)r`r`rv)r`r`rv)r`r`r�)r`r`rv)r`r`rv)r`r`rv)r`r`rv)r`r`rv)r`r`rv)r`r`r�)r`r`r�)r`r`rv)r`r`rv)r`r`rv)r`r`r�)r`r`ra)r`r`ra)r`r`rv)r`r`rv)r`r`rv)r`r`rv)r`r`rv)r`r`rv)r`r`rv)r`r`rv)r`r`rf)r`r`ra)r`r`ra)r`r`)r`r`rv)r`r`rv)r`r`r�rf)r`r`r�rf)r`r`r�rf)r`r`r�rf)r`r`r�re)r`r`r�rf)r`r`rv)r`r`r�)r`r`rv)r`r`ra)r`r`)r`r`)r`r`)r`r`)��os�models.arch.ablationsrrrrrrr�models.arch.finet_baser12�models.arch.finet_base_r2rrr
r�models.arch.finet_light_spy_br2r�#models.arch.finet_light_spy_br2_decr�$models.arch.finet_light_spy_br2_dec2r�models.arch.finet_light_spy_cr2r�models.arch.finet_light_spy_r2rrr�models.arch.finet_mugi_r2rr�models.arch.finet_v1r�models.arch.finet_lightr�models.arch.finet_v2r!�models.arch.finet_v2_2r"r#�models.arch.finet_v2_3r%�models.arch.finet_v2_4r'�models.arch.finet_v2_6r)�models.arch.finet_v3r+�models.arch.ytmt_nafnet2r,r-r.r/r0r1r2�models.arch.ytmt_nafnet3r3r4r5�models.arch.ytmt_nafnet_addr7�models.arch.ytmt_nafnet_concatr9�#models.arch.ytmt_nafnet_concat_maskr;�models.arch.ytmt_nafnet_fexr=�&models.arch.ytmt_nafnet_maskenc_concatr?�/models.arch.ytmt_nafnet_maskenc_concat_maskgaterA�(models.arch.ytmt_nafnet_maskenc_maskgaterC�,models.arch.ytmt_nafnet_maskenc_mix_maskgaterDrE�models.arch.ytmt_networks�models.arch.unet�models.arch.dncnn�models.arch.ytmt_nafnet�models.arch.ytmt_nafnet_relu�models.arch.ytmt_restormerrGrH�models.networksrIrUrXr[r^r_rprrrsrtrurxryrzr{r}r�r�r�r�r�r�r�r�r�r�r�r�r�r�r�r�r�r�r�r�r�r�r�r�r�r�r�r�r�r�r�r�r�r�r�r�r�r�r�r�r�r�r�r�r�r�r�r�r�r�r�r�r�r��__name__�toolsr��time�torch�ones�cuda�x�net�randn�data�s�range�_�e�printrSrSrSrT�<module>s�$$131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263
64
65
66
67
68
697071727374757677
78