From 179ae29aa2c4fd3ca4e58a6ecb6b163b1ab3724f Mon Sep 17 00:00:00 2001 From: ChairC <974833488@qq.com> Date: Sun, 17 Nov 2024 22:55:29 +0800 Subject: [PATCH 1/3] Update: Update organization logo. --- README.md | 2 +- README_zh.md | 2 +- assets/Pycharm.png | Bin 10112 -> 0 bytes assets/Python.png | Bin 6176 -> 0 bytes assets/jetbrains.svg | 13 +++++++++++++ assets/pytorch.png | Bin 7617 -> 0 bytes 6 files changed, 15 insertions(+), 2 deletions(-) delete mode 100644 assets/Pycharm.png delete mode 100644 assets/Python.png create mode 100644 assets/jetbrains.svg delete mode 100644 assets/pytorch.png diff --git a/README.md b/README.md index f81044e..afec445 100644 --- a/README.md +++ b/README.md @@ -493,4 +493,4 @@ If this project is used for experiments in an academic paper, where possible ple [@Pytorch](https://pytorch.org/) -PycharmPythonpytorch \ No newline at end of file +![JetBrains logo](assets/jetbrains.svg) \ No newline at end of file diff --git a/README_zh.md b/README_zh.md index a34abd3..5fb06de 100644 --- a/README_zh.md +++ b/README_zh.md @@ -494,5 +494,5 @@ Integrated Design Diffusion Model [@Pytorch](https://pytorch.org/) -PycharmPythonpytorch +![JetBrains logo](assets/jetbrains.svg) diff --git a/assets/Pycharm.png b/assets/Pycharm.png deleted file mode 100644 index 2d2dfa984ccdf463f2ad04f911790e0f8a039906..0000000000000000000000000000000000000000 GIT binary patch literal 0 HcmV?d00001 literal 10112 zcmYLvbyQT}_ck%$AObUhbPOpCN`nYPHw-C5Nq0&&1HvFJA>Ca9(k#nm;?7h#kpS$;oP*;`1drt8j4Gj%XL0(z|)eipm0I^Z;%J)@Ss0RI` zhMXi?)i~8Y8X5#mK^m&%Wpr$a!Q;p!YOD?I^baveu9!?p=T=B$dM%jknXP4{!QeRaH@v%RqIxOHmfojh z`mas{b`GzEud+U#oOT$Wc#Jm#imk&WPTf6E6P6Z)f>XGnf_Cj$vM_wZtALeE1ADousj?XR(#qwiC zkBlUx`|2M&+_AJ}YjY>?p;|GUGN%N>Mmy)G5j@c%LwSo*oW+j!h)TGIxCI zv)jG&{M4<4_U~oqYkCX!9l_bs&^e7UDg+A^{`qV9+_U-*2GXJ9uIapg1G|N%WQ=Kf z+}oE0JW6zbU*ThKSOk5BA6ZDNR-xX}ej-HzafF+FV15=%~4LfZJO8bTskEsv^WYJR-R;O6H+iYmM zAZw>e?3f?yjMB3HdI5haD$OZ4i|hIYNuuPoE)8J3DL1k155>k-bt*o4AAw)cW~aK2 zws7JjK4f6gH_~)rT7Pi1Fzmct^hj4x#Ozply4uiUUhza%Iv6h2-V067HKj)I^{m)U zXN=#7e;NQy1JIsP@;^5fh-g|d>kwI&?SzSFA8sg?@~wL(BDs`ULuzsxp6WDaeK=;S zdx#3Y1zw_aeOCSwq=d%CXJMZbg7&DcZ*+>K9j07E7d_Qsyq0jfO5+!U3@iQ~8|-Y@ zY1r(MPEK@DZblgTw84f`Aufer6j*sxOzE5-V0FeuKcDg|K=U2HVlkCiyY9qsJn4Uy zh7uPo3j8_&3dgqn-?% zkC!dA!#?frN~S1I>w6P#rjR$SzaFQrrbblrw}0JAu~lZyCmm4szjl5Ze98ZuPxFH3 zTSz2VnV9u2*UDp-i@V=o=yyuz%WaQx_SVN&mB)MBa+`lCtJ`W{ESldk1jJ_$yn2a zV+1Hzln%^{Na+k}bNz1bY>y*bj>A3LmAlSuEg0D~X>=&1iA)t83bLNGTQA%p~uHml)E|81< zmV*mfE_nrVj!O!v4dFBuK>sV%dg-ikheB5<85UTdpt@bl62H>YHneT#4|Y_+f7 zwU_1rONT`7I}S_{qKmgA?1jl8+aX9k+!KZ(u zENRQNk^_1EDxtt4+a}DB)6G^61pe0c>!XN_Ge1$L{kE(+E`*Bw76Qp*JRB~jEsAw} zD6~`tfeXJ<;uu3mx1d@TokKC$>2%dA5*fB2oF{^YVZWtS@EJIDME@Z3j7+mQ5>Ef&e_u2-cn=vP z3g_#4$K#V8fD6JeOWz;>ZTXBpH5bJTR0-;3yQ)g%lL*ag0WJD=XPZ06@D7@bk09IVDT{O4+)h5UGK=OdwY4Ho4t-(eT4A1)s zs*%2L>it4n zy((*g1?_nDQ_87;3b7Ym&9t+W4Jkx7?nMd_ghP_=gDwPyR{>zXYgc;Vr&&rSx-~i^ zm_=VTyd9@jE!rt4{X7?3EdqFrVKK5_EeQRL`CaW}u<>gQop6Y3j=eDTC9`~L;H3~q z+!j!Q4~>#obQL*L1)Xw2ZbSaRoG=t)^$tDF`u}w?74dmFG2m%zX!p8&yK_gER$!SFQj*K zOu9#7;+`~qcCW?JIJl~0=njx2I|20q6Dtd)K)wEpDzJRAYdhx;KZHOLjbtjEsBXWn z$dfCec+gcp%{z5OSQxhbQUmAh>27vlzu=l*NaLyxF1tf|tMSxRFoHkM)Vvq8{r@PMy(@cPFi<)i%@26?)&XC&aD{{&lA6nc_^@59s6N zB6wZsQ*RZkI1tFM+0V(uDY0@W5Io;uzc-4r0}znyJekO|f75D|MS@0?Kc3M>tvy~g z3hicmJWF4@XJ!!9sWONlHS$9L)_zIx?(upn%DmJjBM7p2(I+CWHOW)$Tu+e-uZx!f z#Yc&RdqtBx-x;hHftub`+s?8ryN+;rUCiq4xTz$F3{ZPgwHa|^<5QJHw0Y5G`~AVG zYWt(eVO({id8BZxA`hxrHeCe=ACCT9=PKHkjFXA;*O{Q%j;S^&#`U0=A3=7aw%K&O z8Che}gKhW@PUqGYgq|Y&kvR;Xx-32B{f`Opr?*9_nV&BX=MSs6$)Ta9*r5qWiDA%B zS9^}~85nv1*78(;u9PQ}L>CW6WS8aL<#7t14Ixe7y^Bh&YwJoAY66!r{&{OQ-I(hB ze%bqdwZQI7&((-Z{l;ONyfRfbL>iaW?Ad%Cl9$gYkwW-nMH?^BJf~nOa!kLgy;= z!!Zfos%P}FAWj;V9DArefoo!RBrX*k0%vfNLO_HnV^EWPo3K#H&*DPPn%eC#NWkfU zbmAhk6+vT;3zhyl5JevHD$7gK<5Vt+%w#XV#=a+wT}MafK+=dGU{fR3;(e+mM1olY zvk9D<{VEq1bV`ZgQ%qDed+I~9{e9wHPpo*+>X!S-GWp$Ur+ACUaT=^~uF8-yQ)W9+ zia?B**6$4c&)>g0T59@o_7#EG8_$k@)#IWJ$?bD_l-i(D_FPTuLhB!ke(V<jK_(>K5Cz<)P!JkMKpWE+xg+MK_5M9F;NjL}rUce>33p#zu*&jg zHuBjoObnUJmm@$f9lf6i73OoqnDmWxEG$I&re04;-ZI1pVVD+)>?CTctE)ql#!zb| zaDkTU4kbdyJRHpPa(@J4rL{}MB}3Zh#umS8nr(}=2PrVlD*__wole`Y`rEHoBma(z zs)4Cu-~N&RXDOl$roFh?&Hg6usATb4YBc3Frss+4gt z7bpuJvgy!FOE4gUesqE+KjC5G8jts8B{$gMXz1|nf}G8hNQSqEtJ$FNBjf51=v+^Q7i>1NKo_KaH2Wdc#s6p{JW=gCkKSR1@ zV;*B2aq-zw;!DR+15=#gGsLxB$QRA=H8zSsMcfZUXUlcj;rI^+x}Au%Q#Itz5$;YZ zF%fNeU0WP+OP}lEqE)uT)-e5v*uS&HnA7rPfKB{ZSKhi2R=x`wKTDs^V9;|Bn+KcP}lIE|0I$5kkM%UNLDTx2o$fxwT zBk$1BYzSm>G1S`vG{sY&e)W$Rxh5D5G!nz6PgNEL6;xEbFb-gS!b>Rs)**5*O>8_f zO0yYev^;aDrz*0~*mmAe=~HISpwErqV_zhTF91gmxWkSxAmNa|NasuwdGdAJylrB3 z?aq)N5LZ`Mb0~vjzf(tZWn+xc_{$T@=H}*FbEKl_H~LF}q^H2kOdNW@rUQlU8%4p> zeH_9wE=b%xbt!hfWjMLCo940$vXxMHXCmL8dFGE`nZ^-*7(Vn-CWmPI`AlAxA&|aX=W6+*nw}xa$${YN!lerrG!3P1 z107$OLdN?{k6$*mhfAe-FaS7(XR9!$bzPE5sddVPF&u)9aiR1|eq`-f0_2i=bun)| za?*g>qk{d-bho&vq?_{ME}`z>Hk;{T8GK}yhT97i4vJM)bq#S!kDiuqT(`B0bJMlIE$G} zma4;qd{2TC$W-XKwK-=f#L=IOLLP$6KkAOcH`@t#?M(@GUv(}jB;r^UdVh-zh*8OT ze(oqrPJ{4wocDUEvxByWYo0?xCz6npLfdgtkc^v|7T>FbS=@9%C&y`juyi3~!)7Rv zD$>$@bk6i8f^ULR|Gld=a?CeP4EgN~+TvF>I9}E&U~*Vh*frF*ZNaIR6zs=ns=Q5N z6~RQx>TvM-I%YkG{`}z;cOm+z6hM)BFeh5fq2NDk=F2`?{JfIU=%xjdQ$S0maak96 zk!c%0mj?BctJX9zev#v_ZyEO$C?w_nOaxz<}HYxe`O--2Q?zoXwdX4&64 zaredNVs#AAfSaAKd%Qr?znp@w3&!cpv!w_+8Sw>&q~$S3ZeE{nouBm(l*6p{1rQLi z08czw_rPanK5a3^o{F@JkT)%tdN*VZpG2*e8%wz7*1J7$ekDvm&`()#%!ixvv~lP+ z#ihdtYH&zvwl`P0l|K(`MOcbyX*Ad`68bx;C~$lxyy0}0k@Y?o$cI^MZxfwgM`3I$ z(8;jLRkqS_JBTU5(yxbyEMsT>qcnOL_0eZTgD-BPFPM={YVwUx651*xsgrlFX0KU+ z9Ybe$s~y`|4HFziOp0N;zJ=Rg?6QH_M1g?bAw9p8Y6Eh`!xfJabkI{`vYa5n4|mFb zR)@KwcbKc|Io(CCJhl}7koTUZbApS^3-m0^+Hvxa{x&M1t(q7+a7pW>F(<_2-PN4yOzqXOrZg0KwZ2uaN!*Xb55HuVpKus zg+h8hJ4w1f+kHLp;>VO{L{Hm@+I_(>sqnALIJLYPkz(SiPh9O^78y_m-n=s80@oRu zvq{X*$rHWbywT z72r%+P+PywuVm!n#)Tg0REQ&4RfAtRp2YD7Gg0&>I?~qF@}7)@>-*YukiiP4@&2gj zp|x0IJ>rn&z09Mzx3D!r!UB%$L>I24>wp%^kQ<6L5=N4yrBSV z4JSFYmtpBv>~m(fBq>?(+egp0d)g%BvtS@0?h^mJ0g=CruKn+NwvlAcNUmKHPPn<3 z^*0g#SSX|QQdaAMFGq}V^osulmxkX;Rve@-U`>R6FcxdpHWbZZx}HS;>7U3+MvYGV zOpI(CiF5BAy=fqQ@&~D~7s1HaFpW^%YRISSSMxG55q{0?_4I?7YuX=%bseb1XUA!} zFMCle(Mqv4ETCERO!D+)lihE49{^k`jG{qm=UcHwp-U+0r>}g@)C;Cdi$Yk*TqD6^h)pr*B9C)s_AAA!2XT5 z#aFJX`(rp}oNCs4c>AwihNJ8|r9>)4t2nnmnQN8NK{^0%JBQ&7PY1zE8PSZPqj+cR z{Z_}hN8#u%wHMLY3&!*HBt)(J4>BKtdFVL;M7d1R{#n&t1T8i{Ll*(x3%0Zpk{AzO z`E%P-3Ej|D7`wWPXic3@>D6=ju{j|CEF`ae$yy<r!4Qgn0g1wuNjn zWj>Cuwayl{2kHgwaTZD|A?zJFq!(YsqV*C8yx>TSQ;=I4gIJ)SG6wy&qSsPbI&Afi zz%g7Z@^S&@0e&2nlrupA^m2XzjV>=qN!i=4kaE0Ws3pb>mv|ga52&-w)eJj=k z+5Pyw`=@feq6BEoPFjfnzrJffkCmcg_2I(2@**>}SDuwP7-y=D#c*$B7pmT46xptX ztUHT^Z_YQ1Z!NB}>#J2}kV~u89nJ4F2S$2i)euF|$vIOMi(+QV0v*y_+`OG*Is`I$ zucI1J??+nz$h4D+Gdlczv{?H3Po@#10JS<_PQBUS zoNj%Pt-kw>AzT(m8CPrwXO;EY!mO3zx+-#m>ZB{HwT_F9YiPDUZ(9($ zCcpQI3aYm_kOOsy4rpcyJZZmi`$gD$N{yK}P#>njQJHnvPMG*drgsHSjj$HOnEP5x z3$11w&~#6M12&W2cd&l2PV?jcN?&r4a620y!qlHP|pT&n*d(2HHKyn5@owW7g&O%7#Kx@+g>r4BRslp^F z5WiFkwSyZI@GNCXv+#ht%nQ*5?TUa)R9V{WfTVG^k6ER35j-8PyShUY_SBdr_k`>c zIRS79Kcy@8rULbq5xH?@!$@%T*B6F3^p|w&4?~*mMyzZOIGa}vOp>~VlADLLS&ewO zSC7({rAglvCWu|`8t2m?fGAJzGUCqc_H>BBK3DmW(vhCmyQrxU)3H2xa7;E1?x=O&Ebw2C^}tHI8H-jP(U zrXv;(sr@iqrq~Y#tY+M^d4KcoxAf$`@O@H|mScrJSWoD@t+^^zOdBja1WA)9>uC=O zA33^ zAe5d=hw%DT^xt58Hv0ZWEDaxOE|tXpGMWMTUQlM-WtTI+2;U2aNSqr#RJ{06=}LhV z@!_m|BiyI)r#o>ufCA>!6Nbma$;mI7ROn8o2s#UKRI9Hjm_vzFoF^%n4EhkOPyFY0 zN`XBs6jqE-viIp*r6TI8;c3ACapN|+;vb_lm&gE@6O5nL1%$foecphjLU_3m$G>lfs+3{ zsDITg_iDEl*dM21Tv6A*^W4HgV4d8T!S>Z{n33~x(wdsuS+n1<8Cv&}-D;w3QH9Si z&bt7?3`pGWjh-| z|CTIXd{l0Q4F$#zaZPL{>PNCbhcVv8$6m!rSKs-}tRG>RCF`Vu81LSt#>_JCq97(| zNmGpe<zc*F&SPe(mWn*zZ{mYkItV#w09lG5QZ2H@(%&^HmT z|431`f0^_>TnUT!E6TN%A7oXu9Z8^RM!Bw};P$CUG1m>-|4JEeA_@oKKbu>NX|3FD zR<)`#n37A+(({(zB9JG4$Q{)#*ds(#nb1$i2(62*Cyxq~ctcYDIKU-KQvoI)0%D2} zewhPOxr3QsDn!GK*?CnfH}O7Nf|UHehf$btK^L;ON*}c)5s6rSk!VpF^V6}@$|&>m zB6HUY3KLTmK`j*X2uuK49!XlT#4{<{nCrt>P3F0=84U-dj`GWuvLl!KQz zjNDWJDiJ@40z|cb&lG%P|J~a{fAL?9Kgxwfz0a5 zKDjFHa^C15mg$g5vOQ$_91?M3D33Eb22Q;!(&db!(&=r(rx)_Jd6Q?Rm=f`6UT;j?q)GvRX3L=B*0WN*G=s$AlJoo z{yI^A<#r05uIos25q%SdP?svvsF*+-uJL}tLc!%iAjtg?9qj17qdjH~L#03~N2+d3 z?!xKB06&LX^mc(BARcP;02BbegT02rDMAm}$IQtg;Rc7J=76`L!^>Er>zsTch?{7RY)Kh}u?&129ov$ws<0`gH+gPh-H z8!l*;h+oB^;bnU{L`aYeu;~xNu#a7foU?{7b>wXU%S_zIXXr%eu9WkIZfO}i+9IJD zYi1xwM^~oqL)348jiJY^E)}tC#&InzFmy+1L(c^fi7Vc zZ2*7B3E-zn=R4+v(GQ?O7b89zpLA>UGtwfIg#+Iduq?;16V6AI!eT8y%wr&Ol*7ic zEK^SRW=Mi7pM2SOy=A>}MV*Eyi--!`7iW3e0$C7pNERD<69*T0&ggstn>4eGFUP<$ zd>$iglU#PEzovOFE1d$~JN}cxd$~(bKZcdNK-eP*%L1yQTe;Ts23cDG<}tOfK6AM3 zIQk~-g&0OK>c>D-MyiP3=H9AKM?j2L>$Mc=YZgf<{cARgi!wVk?L*Kax1B~HsPSLz zVpEaqLAVi?_g-vn)2Q)QTT?@e6O{~1{=L_ri7%pqf`$y1Xx1?+OdZ0kxIm>}eY zCF1o?0c=*X`LaZWi2Hj0|Gv{a!XlHK|CEL0yIj+Of1AVuwLu#TM?hEu%cjk!>F+=# zY23oMZ>6;A6&$$jp>ahP^UYQAJu&5@q9m#>USpzrX4y&HV{(j2iH3(%X?Bre=XAT& z!|o!6od&wtBIt1o4Q$Y1v)DJB3G;P$P`DKlO@mw$G!5tHDVjVyCpHJIx^ojZOIx7*H%Ptx9)5AR@{yp_&>=;e+Oor(**cp@oFLVIcP+^cqLS$W5D zYIRj^Xc_o(idi@ImNa5@IUa|^r+y5MI5P6cTxD1WWnrzqnYHuA8ff;xOQdQR5K<&* z{N;qG`MDFH#pI*nH4bkv7yIu3s=LUu(2Gfi<`c&t54H?G`S4XbX6Bn)TJo*ghLaF- zu2k5_xs%M90H+XUCApiX(u2?PR^L?5{u9y-d$&?X=ASsao>))Q8zsEKp0>{Q;9-EI|k+57O~ufh~$&H zX^8_1yMHXrP!aODP+eCpUhcu6Zvj%={yGVFPn)ikEqmeiiKr`eXbLi_(p8cl5dQ~| CP#W3* diff --git a/assets/Python.png b/assets/Python.png deleted file mode 100644 index 8b1ffa0a86957735bd676958decddb53771d7376..0000000000000000000000000000000000000000 GIT binary patch literal 0 HcmV?d00001 literal 6176 zcmb7IXIE3t*QEpq(h_V4Inis z;;m|mNfUX18B&L=W}J7AjErMRUl(p3YJXT1QJ8bDZJ_(pN|qoicfpgqMpJ+r)618^ z_2aUKp{nTI_bYGQ^PK1OJKvA$T?X23FV2Ok{W6Q7OoAdlWTw@bc9-zk=+T z!cD3ZIkCaf=6OCQ&0**sCnXJ;f{^~ zdbr_{43HiZ3r*hRrzS^$rfKLYzJp|7$!cHHEDS(00Db)zMg^od0KcT?lLoXDi4bw| zM*V3%(!+OOc^h%~dytTDf+D&d@gCI2=dC9x4CKT-zPB47dE|s}Q`i;ze51(%5y+t7 zV>{L({%K)O9z%dS0yk)~10*gcC69!Azq@-W)Ts&19o4+|e$OA>3jZmLn*~ly!h zV!*aezQ#Z2&yly?QI;YcX3qepuCn&{Bc_Yh{w*dW9=62?aoYM++LAT7|4ivJgdxk4 zEQJ%}$?(WFjCwFoH=T@&ooxG9u2Ma1kZnfs{`g-Oi;t46yJfO~`bQI2S?ht_;oLHG z9}3jU--!#)b^l^u9}5*PBI~^AfxtPnF>LZWBB#a4wp6mYMnA2(5o1r8U5vLtkHXmm zosE#YOyot#O5fn6dVNv2O&*_RNoj#1fFfLUu~>XmvBa z9Vz_SL14OK*6W%=6!NNIlj{48jX+R&>jCS;2rt4VgPicW1?)%5u_F+b8eOkWwMl2X z`Xn>=mlUj-2XP@=TMk_(!SVj0J>6(23SEe)K!|t2e2zg|T2Sb4P5c zIvg48^ZCT~T6@uct3P{vWc%0(2ExndP{m z{-xglaf@6Es*Cr#h6V}$nxwmPTUL3WCfLXa%Ox|6WbU$t)t`EDZ&?7F)j@sCPyyE)w-m*iPo8rtrn22|OTADBQ7)=<(zJ#X_S?)PRRWYM~)23vE ztFz|ebg*&V^;IAy5yD{l(P`@<_QyYe9uwKWtbBjesBb;6a8)ZfwT%+xFJ2*YS221Z z`>ONOa@QC?psh&-OTLZiwCC+A^gN{4=Kt5yGE+q`s11Nuf4w!mKo4hj2x@^1&ZS76 z#c2ZPtXmDq>Vs*ROw=`3m}LORi4OFQ%Z5|VpkP5AnE;$_Hkt@wfMX}uub+4krNFw9|+zZIe+x)^>5kSlM2!UIxV;E~70Zn-@fFv*XToN#TiIm}qc&8q* zXC=3|DYcoh!o{t)!uKo)bi~O`hXsjQ%w$An_qYL4m@l)>6#*&t)_G#2Evna=!D-NB z23!w|EP$5DBchldez4A(nGHpuv_A(s1D$x@$^aPPJ=%*YObC#uJJk&y2TWm8Bu(?4 z3UGl9f0)(~wdUUU0p37B3lMMCYK}7o2F%6cu2q>{#SvM#iZf}|S!jN@V4?N^v|YFi z*RvxekTy1J7W->lA%>-3RsMF+#o?nh7@k-vVWu=B*(zfeeLjYqAQ)Sb-8D?Zlsi3mvpb>%UhX;$L5 z>EizNG898To`h%{|1hH|8kOxDhZ4f3E% zBvfwDT}xO4%01k>IJf9Kq834qv<2oUXu|g&t=Wq8DO1io0()FJ$2m z&8*Yw=Le zP#}u<`;>Sl*!nyXxCar|=GIQmQZlORb8nbsb z_|)pLOy2RDlkcMDM*wuKC!$3&eQ*-HjEK@sM|Z-6%6z z6>|iT-RHOWz&S?!QFw*Eea;7e`?IKdUa#PO?vUKdXR+-f$j4#nIG7++k8(PYTqDhF z{X03(pV9Y@Zk$(Ol~gACIFUJC31jup@}`$n{Mr~ck3P*?0VuW~h5<{BoH1&HJdzH@ zCjn>e^Y@}W1dmRb<6E%HHdlgNe`@l`(!6!Rrd>JMo)|{fu3DaI{RY%Y8BH+EMa z0J`c6mI!Vce`@~{tek2_SnX9ZN534F1p1`H6K@L*yk72KFNPdyaT%XgEpHzWUq7Xt zyKBG27Q8(b*J}T?BqZ^{L8;NvE$0VSwD?c*~^u@&|@Cy{OsPRl`tCF1741ct{GGBB|*7(@R zK;FU6d65pjAxj&O)iik=8#`WM-WsayuiK@)5aZa5<3+aVFY&>bjy#zw@6* zAisdi@eRNS6;;|#%}zk3mjJ$m5_$vdZpaZMw#vVv(KqET(T=&V&}M^mS9jQ_Q}A#X ziewZ7Urmd<7RB zTUuq&!Nr`z%Go2FJU#a86l`-84!$>_F#o01|i;<3iTH(jx_>P^nWneG70=-UIt zT|##P|kQhe!d7P|TF zs2aTG^DpVccat^a-KrxxOKy3Xz@GhtQmvQp;Ah=4W9n;mR0vT$T3wuTIcY9@A<)Lq zdY1N@x-x}#{V5JR0Cb19pjpKGl&ro%@4xA-g*JyAQu!T-VhbDD7FjMN|4VmMDlzyeCQLpj&x zQgOpXFs<%Irb5PK=2vS`xS9amrgk;e@MyJAImOL~M(1V?s z#Ef(f3a5LM4+5%9TO6d$L^eHoY{iV;G3J@7k+}N7sH$6;AtJPvj}Ml`31}A{P%G5i z$3jT#t?pYp4yAixRqxUgM-|KS>&T+y`Wr=+g;=6qMRxq!9VvW~@U`&R4UqNFk`v?I8ZP48H+M)xigzV_z1SRG0D`T>U$JPbPc>4I>JpBN2P`VcyGC;ArhiWuklMlkS!M)={;Q2HHX zeu`m}R|#dyfbYc+udP}^r!Mj*$h#V(_8XhDh;Gu{soUnSc*Q`zrH+2FY_#;9R_=AA zvP4L$A9u1bi(0^)t3TvOoW$H!eHX>CS6V+0`8o={9jpXz7e!s3KeX3bn0x9R6Le~l zeT{}ehjBmgh2-T=bhR)AxhuYr7~E3dE}FbmkN20~hzy$Ys-xWBBawD;ANg5{x%y`{ z#qi7YSu1+3xP9TB=sTZzNSt?Co514V;}c~0)n$99mfIJw*Visl1Mc=o<~;`hh}J`G zG2ui#oORCf(22|D8SAfM8f%%umP@wV9m=fulD`jaxWSzLHyW5ewHYn*z`vzi%PJ4r zgog<1uF2!)j=m{c-fTUD4$XB~`}2MkraA(M-%#T^McQMC)oPsZc4dw&ollx+ahp)w z`d7KQk}%pYYH+uE&1JOv<%C^}S^0yHEhjNYXW*(~3ddXyT$(op*t9sCedX|`{}G^G zP|Sjz;$b*Le^YexnZ~FFVfARUY50sKrZ~&P(F`BjyqUklMqUDJ2&8v;sdl{&)aRk) zu71f~<|6p-q#$1lP?;{cwFN0T<;F=~exP+Sn!u;DqxayBBm(slX7>;1RCI1X5_s!m zcplW3PMF|S%x;R>p*h+c*nL30Apn>1@XB%5{CE?9P8&Z&vhyp7{MuP@z>Kyqfr@i7Pc&SW?2je6Vrb0-ZsiuGy0`8##OgH z_ONMHmKuS3S{o%zR?mn661XscbiYCkAQB}X7um_d3pahydhjpS1c1dqiw7|JrSXmC z@WvHQvP0nK5U*}0&Sltl?a-J~vgcqS(@Z)&tE?d^DeIe;daC z`G>}07PNMPPA+Mewn3%iU5$Xz-J|1Vewq_FQvW%)4}4#i%|6K=#ueQM@MbvwiMONQ zK;W*!iUYo?tu-Ou?8TKL1b)xG8`B{%s)Vad<1b2k!(qXxdA@B;Kxs%Puv%n9z=cXfkZq>c3HQL!7MtjIRTk#!}F2Vt+Oz<<_z-hxt`j2%5ltz+$*GT%Wl$hxF1tesOV}Mg6 zC;`5vd>-SW7crp>p-d-_fM0To?9Y4HN~sXmW>JMdr-|&6W0rMxDZXtAKc3^&URcx< z+Um!ti6;efAffStW)@osBxviNoMDDrv?i-wI|jD5rT2IpG4$(42m-`W`q4o6RRE4f zXrL*W?uDHpqH-Z;RcCw@em|C7GG_*?YklK7?@l9dX~-Y&I>eA0qfDjheNq|GH!Ncz zQa6R>Ys^XkGAYlV0$MNXH%_CsM(wrr`KHbBGT$55q z?iD9Hvc~}?`lozr)I+i#y*O>Q*w@59pZ6%eYOAmJio^t-v=RP2(Q33ib)ma5_5aTF zX2LiZQn*P*v3djuTvKD6>=zwG@ajsRC1Tu8jlFYu2U;0=kAW|FZ(RR7jPDlOy{Qhb z7Ie?qi7U!91*OYs!c>9-JP=jY6v2)7)xJUHVpmtSXdVn2Iw4)pvP>cS1EN4>uM2CT|YE6B^LjSr@(y zV*@On=DXrqO>SF+-A6_gY_`tSGzU*UnR;N!?;$zjt2dH=UjKq&c?y9WKfk!_7VN~o zkZe~`d0|(Oq`6~RWo2nv_!X2DAUtqCNi~JlgaQ7o3?=FXe5GqKY&Q{-u_Y;2?BIhZ zptCk^0kilDqP6`E{oiC+9h-w)SoDX1fBGy&jn;;ooMX_2Bq2>Z8eRT_-9LmmH4As9c zd)im|Iur7|s{GzIQaVsT2YV)>=D0=`cy0uOAl{;(`;;Ialbl9?Qd>O9a4J zUh%YMN~O9)KKfO)xQ!(l9B&$j&9gs2BEKYtn(0Qz*X3v2myS4jWQJi>Ge&@Dl07)e z-pEp}!+y^0mPa|i{9AKm{?YcrVmfcI?_4~!f+3D%A1B>-Nm8BsnvaRl#w!o6ugQ09 zCpqR1U0REzLKMyijNoKGzzx`267PGAxWh=nsJ*y4DGpt63ZvcCZA+)`erOvlenBi$ z%F_D`5?a$VU7c&dQi6X~`mlcatwzlmU)M!>Hs`Vw-f?=c-#x4qOXL3XcK;70n&z}O zMkE{ClY%XKp(1Pym<rL^s-~3E34{gSmiNG%_#7}||8uyZ)Nyc9W0GC&g*XD8 z&ST;4`6%4}U&eNM`Gkm32+84Y?HzM}1t3S@vRRkP_ZUgec}Oc4J{v;(7ri?P;{yw? RlPrHS{aePmH9C%%{{h)iRm1=Q diff --git a/assets/jetbrains.svg b/assets/jetbrains.svg new file mode 100644 index 0000000..cb3a2a0 --- /dev/null +++ b/assets/jetbrains.svg @@ -0,0 +1,13 @@ + + + + + + + + + + + + + diff --git a/assets/pytorch.png b/assets/pytorch.png deleted file mode 100644 index 1c550d91198671ddde645b87300b4946a61c996f..0000000000000000000000000000000000000000 GIT binary patch literal 0 HcmV?d00001 literal 7617 zcmX9@cRZWl_m3oqo!DFKP0^1%g4Cu(?NNKw9xZAmW~dQW)E=QITB}5BQ=?XCwf5et zR?*V-mwtbLJkNdI=Y8($-g};V?m6#slI|I4Q<1Zh0{{RjT^*zep&k17Ku8GhI?LBZ zgoY@{L|X&!dYpY10N@DMMXKKqwLi4;Lb01=scv5wEODDoqgJ`~;Tw}?FsV~>3?J6z z9W4=)y1DH=D=@mG3{_Qf0Q{J&9Kj=057CQ=D}SiL(};rJgU*!KWICXZp^%yB%n)fA zJ9kr+Tt3Ro+@se)H8ty!|mqaLw5&ez|$*e zhuCed{BuwM0Aur(T1tbAPu7))AItouOq*R@NDnS0G$g?-gudU#`94{bto-ALoxToQ z9R{x{8+JXonb?4@ttk3_wLoMN0o41jGgwCeA2uol5Qda8gYfr(NqB+Z1dwmd>G_Hg zSvmL_K7o>tnjb3h@~8iE#1;O4shXZ3jl7>Cy&!~su5MnTsn3u|t5ZQa(qszE81#X) zu@i6|2`%DdJval_{FQFo+C89f*6hxJFj{Br9qm6kDb==ZISvj%{2PNz;=>x|91+5O5 zi#9(sKbAHbb^_(c5<${U2&;fi{GnYCMxSe%eEGLk#{n&6B$bIj0tLx3g10)l~BiEkU?( zMEQ7Xz!eo5S#BD;QXw!Iu)#@Tf8|~t6`MdSlj^Mgi5Wh_1%6H1I~_|`I6+n1b(&h` zoK&O6k3Gz)PL>DE(G~t##WA{C-1$#^&Ss3}I_5PH;nk{|WUpTGpgCAJvTJfe#282x zrCZ~_2}h0H6Dq*k;%};}`d6(6d;p=o)qfx>K4PD$e`#DMnfUDa*!EGBi>$$CFgp{U z4iDCi4TKb*{4Fh7ItB0)SSG3QUBzv=L_J9IshhvujCj=m7epCC{+w6!!wl|CNdE-@ z1TB=U$P|Yq*_k${U$0r_Qn8uC?Y;H_p=*k6s@L%pzBSmJ-FKP4gYHKK)eYp_TS;~N z4V%m-gK9l(660d5$v$8+fu{y;TFQAX6cygyVE{7^$O!LhzdN)aiZx>sq7uZuvQy|* z6aF~=YOT&}uAK-fv00ED3r^vbQfo?}lNq<9;>T*W1)dw=ZRaeLFzfPy@LzuvtZgog zj7H_1|4Tuz8nX9f)6qaMu?P%gA0>3aG zAai)CSlXI+c9tRl(cY_evSYva_-`tgToLAxFjk96`_?oB(9wKC!jraD@MNcty-qxc zLeQQp}|L}IjR*e;0uXWq!Mah`WBIh?-*M2i5;A$Rt z6Kb1k*S6XJO{N9U*f7cu@xRZ1p&|=(c=EK+ik$?<=;LJ%-}@+a+)PpA0ZZb?)Gv4A zu9?-(?G^(h-wd@3q&~DJ;KsiW=0D}JE^;4p18(cJ@bDno4O(SP6lZhTwHZji)*_zE ze=Pp*_GUNt(lLyQ=Mw{}iUxIDb-X&2-7+e(8UBjvX&ybsz2Y?h`c2C!Va2%;FF7=_ z%Db$1BCio}o4a$LEC>FC<<2)}aSHQJTWb4`v^oWJm8f=y(TC72GgF}P8fKgC)ppfG zErQB^C{wNH+ZaO1^}Nyyb^fFXv5EDE~-cuk@S4`>n&s znE3?@I$Pv2Lu~;Im9x2iK~6!=AFSbrbpwJCacwAOs{%dt1=EAYP&T*x@AG?oILIy4 zD;ijY>aNQXcK>bQ`H+Bp8~2uY5IO7y2F?5Uwl27xR=Z*}%Z&+x!vXXGt0jeu8@!Xk zd6|U|SoNB5PQuu?O(}ivAzZ8b+qZ93>#>7^=-rUZYdh-Y)POwJM>gdqr%8o~uWyZ* zzwzh&0RqG;_`0REl44^&EIOY{ruixXtSXt@h?ZmXlSo{9{{fz*qQ{_RiyrX_I;!GaVcH)5h@F&&U>KDpaCYM(E`m^Yhel z+(&WhsE@Eg>I|ssk5o3nb>5oRe(uB>NWXr5E1>s|aYk7`ch0U}f5S-cZ(qU*w)2gv zC>_**w|)IWN$=$7$(_hm(GM%?`vdJw!m+XR{H=9!j~14jLQmF6U=6?Gf46j(I5;v6 zm)Rycujs%Uf>zMayblG97#Wse3~ojrAauN`y!NHFLZ;*aa^ zu7|*3E4nl-4{ILL2D^a$L+45%dC@`b%L~{q86c-gnq9SXO&dpr2(a`seAP#GvFNva z1+Cez{G-jPxVKY2<#q~792!cpKFp@6*_viBoh9jQXt!ZSPU&ez*~(oXs~3^@K(Om9 z7pN%f9Tc%3QpjI6A3UgX_g%3)H4l@{Cp+hwZYHRI`4<;qk74yn!NLRE>L_~ojj8H+ z*oW+)+Sm1CEIFGn_g_VJ6~bf|l)LjiYI_u{2igXE$S)SHI~qFZSIS6=`&6UG*6PQ>-V>JbCtRol{^s z5%@g^{r0n|^ZUFHCa=>@BNcd${fy>j^~;3r@Bzt|UTi0_)c{T&xbQP5kNfPM({Q{M zdasvF6K}%`vY(R&LBEUo1oPvnaBcOE{$@~EbeJYRQjpDn@cFS6b9cZ0NLIVhuyI`} zINx}g<5f8@;t0=?JwhVGlmf|KPZy461(q!*iP*sx4=+=*8kZVLs-cs1x`Qe z!6Y7EJNuNUigL^@S16En!H0=w7Oy2$UY%;|G3iJYDpDEi+_+Z0^79Zaz8I(<8q;IZ zxe^0VKg*P|dpUn}x#19k(__N8v*b2fALZoM9j_j1eMi^}xMomKRGKdH&<8$w0Bgrl zom~#L%~D(G>M~$*WIv-EM#$|Kt%semX>#1sI127kMiBvMRx}rrkQ>E0!HuoM-PuvZ zCsSKElz&0WN^JP29w3cpIh}>;F*C87YGIzC5GQv1srhb;s4G*$eRT5#tJ*;2R?Y_2 zq+PBZ*(4)3(rbZlr}Un`(-1l@kxUn698ycC`f@f`o6m6sMT`ymt_u+TQOW|^)=)P( zC(p`9uMj(x>bek>=CZM&eZP5M+WOj7TRhP%yQoH7qtRp&$d z*~vKP9x35;yR**dMdgClj9=&|3-da_hjxtIp3!>~i)5jlVyRi1t+l#b`+J4wVm8n4 zbl6k+5q&l_U_S*uR(uT>sTZ>FK)=VlWQyf@YU?kcx3Y}`_Q@}^C7e}G;yvP>5}$9R zjUF3U*>v7IX9_@74n{7?O48162GLHzD; zd}|}q4qoI+@?CYnpJcY<0%b$6dnu*Yz*=$;;WNj=^2BaC9zFTkM&7T-ig?Yc8U*MU zEzrGmfCY18Hd_pZNel2g#Lr3+q-{x2IYFP}i2rKN^&SM-;%kA!tuMy~xM$d?{jRBh z{_xqJY4|he53BMQafbV`B+{;&Wm$b+N)AsXu0be|;3)uVeTBCHc5mj77;f<74SAHI z*(V_Y_7N5k?1p>S#b~qFg+XUr`Ww7^$+qXPdpAZCca+#)6bp=?0R^`38XL7Ps9sMWmweHoH8=|QHxrcsA^=>7>w!Ta!8 z@H74VXk?t{LWNH7uVLe-jhu3?d!LdL%oh)UM)$K)^*F!FR44*v8o~qZZ@QDq3v-7u zs(LYkpv6=U@c!%$^4xqX;Bs~7#U1$KG6&77T1(D~`@77aKn9*_(1j#?i2AW5>EUDc)N1}`?ULkjt>8o-h| zD-@3YbCe1L^Ef^wgVpIP9#z$oiCkaFVD;HP!%ozfban|IOY@S1u$Xe~_KCg6ZqkyT zLy)6AWV@u)3vb1Hs+Jqv7}OTPV6Xho+{ID;F!$g&l4w5FXNyo0 znP!BC-a}+moP(N<@uE|_u-D=2me?5+#Gl%I<6R+{N|z=GOcs`{TaZL-`M;31@10tr zl2-iSAYxp4i)W=Pkejw=pgfZW#+ZHwt}tSmjw+1}vycW`>=|v%{rh=y4ybEkMxt8X zx!g&DDq%Wvp>!}!($Au~`@u?QKmK2&Tc0f*fWGTg9>HSY*-(ZevEr48B zxnl^jH(&yJ{;hmH@H^CP{Q=R56r$;ZBU><%m&*y&A4Ct#K}3Nd#Ty3qB5!>dc581q zl`%~;)=(Z{;YUXYO^w^rGG_kU6P|>q3nu$w>4M!)qs`eQ^oDXn>?|1Upc?W_;){=5 zaF;kEdc$U{NvC^p;R~#%L06FO;Pigo6^yl~RwpL?Zr1f?b%SulENCruk1t*@9%6U6uG*4l9Q+^%rPC$YeSNtm(hE}n!` z0Ed|Af>3Zy|C~!m?VBEMVXsINOelBm+unV=MW2@P?+#%L;c?ngMRsJ;uH5x%Xo_Cb zF?Z}wCzAA80zh&awY2=}j$B@Ebi2^$sQTDyVANJ14=qxBi0teBh1lyN?V-1@O{0Ra z6`^sDp-Xpx`9`tSohvZUs_R_@E75OZ&-Vx0Y5t5zN3#Kdp@P3o)D`B({M=VrvxHBH z3^32aqzrth@JN1`EE=BJ)v%fA}U9SCKX{`;tKg4V_FWUzd0#UvHmK3(RQP^$u?*R>xZqcL+z?-912g|RuUsOcI8d%~?fICaKaFa##>)4FC$WS)VWSEjjr;Had?x!gH+}#o58tQ41<*j|B`nyI(gfbkQ!9 zi?lDwtW=~@0Hn?n4LVeyKezb{4}2CD^kJA3CWI2L;IGmne66@ho32@CqzqKKf4DN0 ziRc7jo(55D2#rG)5`*yDt{*Qi4+I*{jp!L+|u0v3azuJ!>3$4u!rzdp- zm57y&1$)R&i0-`W{ztr5gbA2d{&=dI3zU4;HY~LE^IWB4tiyo31(-F^-QqWq<9B@# z0qg?h)y*FPqil68uxpBZKAf?TyF(DgZhB(o`NmskWnvNWBO|EV#U;eXZd(3E>SMvY zW>K9>v;GJel=>M=878i$ctpuEL!Oy~{z0&!=?vx@+H0+^YpN=LMaG63Wg{>fj5Rj! zz#_|YK2&FP;e+Y*2FGh={nDKzsO7B9M9gbl#iLf2RnY~1 z%23ZBVYmT#;#BJo=*~>^i{HI!E)LoP*p;z&+j74_MA{#9<01fx_N`6nBRv*XS<(bo zr-tbNbf|WHdKVcYz>)I<`Kt3qr{lw3fKbhUo;0ek4*onjqvt1ORa3*0JCFaLww2Oz zVycqJ=`K8MPGab9>=3*#o%--5)Hw~9SwyN-c5%F%e@%3UmSyJFe8hu98c{$=1u`=* zvEl6xmZ1OwABZ9NBQ}`vQB3m9)Bgm+_Yt}^e@9&Z-OEI4+OA>h%W4K*BdU@V(H09s zVA`8k-akCEQruVH3n1frCx_!lX0O58#wkyZS1VHYiHw&dNMni#&XdA8fFT$0?hQ}% z8+h{H>^$}!YoYFdz4x$cu}iX|)=>qpeBx}hZW_THHJuz8 zW4BS?2+52cA^<&YhxqitC}{(}ROwyU5HW4}meoK6pnyR=?+w+F7`sBmS}Yg&~jVX4Ar@+qHGsXEnz!Wd|a;#NJ46 zKevYhWH%}yel?eYx*_{5BTLL3&-2Xgz~|rad?Q-ftN^n-F{l@S-NLbUAxfRog$e#3 zDjRx)<>c&{Qzv6qbBc0H4So++d;EO$3jv1W})Z5qmBY>+oBRXlxh)SIaP*|ZPahCKd>)wNKSuo0euAKv>%4T>T z+~ND)0=Zt}Cuzo@W6YNwAOUO)!R@5qe+$Cj2yYuIdC*gGUWg(vZb~$h#g7J6Uf<}1 ziNBJ-u07%NSs=x@AHS4^&*5St{-(fuW?Md960~*lKp-^Rm9;Pq2TT>X#1IZ4GYfJ4u?9i%RvsAMmc|8V}I5&M(hE%0p{h zb9)4ui;-4Xj!sB6_b2wTfWNIGf1Ed(-%|%u0ghOFESnLnb{C)*y&F(bwD*Jb1#%2i zg04vUp|bi_7Nh&DK6k*w@%BYMLvhqqX~B{iXzBzb`=vN7sEGrVmg0{%08d5oOZkGj zgLpSeb42$Ly*8fvFTKta?X7J(ux z@q{HgU+5CyK(82RT$oF_GL~?lq7;WJ^Y>3YE&O?2NcJ;e{6zqKA2$+QGUymG1vi;k0ctNlW**7t2+A5&V#m$X?{E zri->I%%^@w`Qg<~mfv%E)lii(zhCr3 z<3#g&o56a&J%I9fmp9O`D^`<#fHuTTi9w-{hbK&#vE70f6cc8F6pA}?gO5z zX7^?by0&Q9$vx6&461*TtPzodF~U3y2_B9(kjA3Eg6Llg9x~iDcX~jTpGT{MxHqmB z$)X<7#p+^qkm>OspxppdL;SM_-9G||x5&G7LhZBrBs^WMZs?HuedwJXy@NwgiWWCD zxOf08cpLT!4UyNfUtUjcI{b_UENOLl1u}C)BHs{glAhbKDm7jva9DVy+Z{*RC zC6bc2;Cd#WBk_h4>*jV?hY<2x(%TuX5&>e``>l)8cSQiBnG1DNb4mAemWF&JynoLO z#n&uVe>i_=4kwzm-;dLU)5-!k?>*i47C8#O^y*a-O!FJk-|T!zY(hWW9IG|C;@EX3 z&a*RX|`IHDflr8jq@2#?eb^m|!lTB9DlkqnG# zw;xZh{V6*1T1H6CYwBd)C5oe(Gjoe1^*-uNQhtKx(;6!2hwE@>YFSkl8 zKEsSNXRH1|JYo@l#e8~B!+s2?FOu;IK?_O9{DK~ix0t?ulO57%)?pa`@szt{nF?md zG#J+B=$iY;X1W6L1TylsW2dAct`rHD10= z{>&P*Vg0w&GLj+S_h>`-qh#&}^DUsu2>$0&FBOQy6Jnyxc#9UMZ&_fMOdSu;pWAy3 zyrJ!8n%0PF)$pWKwVj^I4tsf9czU@ifng}^$M7&|%T-}tR|<*=)LZpSG$5A+Moju; zlvE_5>FeXA>|9UXJ~n%}iWo!q1T|}*D)+d0|0I8dg?lVyr*(&^6BpOeR8Oe!2!wqS zbE8Lj0Y@-Ymv2D!=DxvM>1Si$ZlcD&g38Yg&Pv)u)n)#xn$={3UenI3Y^3SDR?n5*{E1jJVZ!w9P-L{|4AYhdKuTI!Ef;JQfCC z@#oi|?~pu4Fbin-i?+zN4N9pd;ac7Tr2&pWqBL!kBjqtj6WR`jhyo&*AjKR7J5Oru zFSV!)NWMe(q4uiF@mqwn3V2e!44M3?7y%yue&WR3iUTCVC!s)Qc6N1bq~?>fGKzQ6 z@Kr{{j0VC6>n}>=tsa+(I93GKoukbmT;`q>O%u8XrK_cXxt_GZe0kjC>j*VjfbJb5 K)^ From a1e073ad23e78c801408482cf563a7b37b489615 Mon Sep 17 00:00:00 2001 From: bestl1fe <166411263+bestl1fe@users.noreply.github.com> Date: Mon, 18 Nov 2024 22:46:45 +0800 Subject: [PATCH 2/3] Fix: Fix the problem that image mean and std cannot be restored. --- sr/interface.py | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/sr/interface.py b/sr/interface.py index 07928fc..3f9650b 100644 --- a/sr/interface.py +++ b/sr/interface.py @@ -70,17 +70,17 @@ def prepare_image(images): return images -def post_image(images): +def post_image(images, device="cpu"): """ Post images :param images: Images + :param device: CPU or GPU :return: new_images """ - new_images = torch.empty(size=images.shape, dtype=torch.uint8) - for i in range(images.shape[0]): - new_image = (images[i].clamp(-1, 1) + 1) / 2 - new_image = (new_image * 255).to(torch.uint8) - new_images[i] = new_image + mean_tensor = torch.tensor(data=MEAN).view(1, -1, 1, 1).to(device) + std_tensor = torch.tensor(data=STD).view(1, -1, 1, 1).to(device) + new_images = images * std_tensor + mean_tensor + new_images = (new_images * 255).to(torch.uint8) return new_images From d76b741870ce42d65cae7a4375293c0ab71fd45e Mon Sep 17 00:00:00 2001 From: bestl1fe <166411263+bestl1fe@users.noreply.github.com> Date: Mon, 18 Nov 2024 22:58:35 +0800 Subject: [PATCH 3/3] Add the device in sr/train.py --- sr/train.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/sr/train.py b/sr/train.py index 4d2988c..3e8f783 100644 --- a/sr/train.py +++ b/sr/train.py @@ -242,9 +242,9 @@ def train(rank=None, args=None): global_step=epoch * len_val_dataloader + i) val_loss_list.append(val_loss.item()) # Save super resolution image and high resolution image - lr_images = post_image(lr_images) - sr_images = post_image(output) - hr_images = post_image(hr_images) + lr_images = post_image(lr_images, device=device) + sr_images = post_image(output, device=device) + hr_images = post_image(hr_images, device=device) image_name = time.time() for lr_index, lr_image in enumerate(lr_images): save_images(images=lr_image, path=os.path.join(save_val_vis_dir, f"{i}_{image_name}_{lr_index}_lr.jpg"))