From d22d223d6d70ee2307bb738e5d3b85459d315a79 Mon Sep 17 00:00:00 2001 From: shawn-sheep Date: Tue, 4 Jul 2023 15:05:56 +0800 Subject: [PATCH] =?UTF-8?q?=E9=85=8D=E7=BD=AE=E4=BA=86cuda=E7=8E=AF?= =?UTF-8?q?=E5=A2=83=20=09modified:=20=20=20helloworld.py=20=09modified:?= =?UTF-8?q?=20=20=20main=5Fwindows/main=5Fwindow.py=20=09new=20file:=20=20?= =?UTF-8?q?=20nst/1.jpg=20=09modified:=20=20=20nst/Models.py?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- helloworld.py | 23 ++++------------------- main_windows/main_window.py | 2 +- nst/1.jpg | Bin 0 -> 11304 bytes nst/Models.py | 2 +- 4 files changed, 6 insertions(+), 21 deletions(-) create mode 100644 nst/1.jpg diff --git a/helloworld.py b/helloworld.py index fc33b92..f0b818f 100644 --- a/helloworld.py +++ b/helloworld.py @@ -1,21 +1,6 @@ import cv2 +import torch - -def detect_cameras(max_cameras=10): - cameras = [] - for i in range(max_cameras): - cap = cv2.VideoCapture(i) - if cap is None or not cap.isOpened(): - cap.release() - else: - cameras.append(i) - cap.release() - return cameras - - -# available_cameras = detect_cameras() -# print(f'Found {len(available_cameras)} camera(s): {available_cameras}' - -image = cv2.imread("nst/ori/1.jpg") -cv2.imshow("Image", image) -cv2.waitKey(0) \ No newline at end of file +print(torch.__version__) +use_cuda = torch.cuda.is_available() # 是否有可用的gpu +print(use_cuda) diff --git a/main_windows/main_window.py b/main_windows/main_window.py index af10cb5..fff6693 100644 --- a/main_windows/main_window.py +++ b/main_windows/main_window.py @@ -82,7 +82,7 @@ class MainWindow(QMainWindow): def pushButton_8(self): self.sub_window_8_ = sub_window_8.SubWindow() self.sub_window_8_.show() - + def pushButton_10(self): self.sub_window_10_ = sub_window_10.SubWindow() self.sub_window_10_.show() diff --git a/nst/1.jpg b/nst/1.jpg new file mode 100644 index 0000000000000000000000000000000000000000..577b6850fb2c7ed2709336c5698080326b673c67 GIT binary patch literal 11304 zcmbVxcQjnz`|gO|B8ccMh$LF{PDG*xiA0GB6GTKNWAxrah%N|1lzh}gn_w7y^csZ0 zFxoIm7-dEs41PY}?^^f%bN{&azGv-sowLq9=RD8e`|Ri4@4lG2zyhur=^5$)C@25` zipv9VF$2&AT%q{a{`FV>H7d$~Jv9{-B^3=d4b6WFEj=9#Ej=v_4ILvLJ;T3tdB()Z z!1V9Mzc2akqgSXYDXAG~X=wjb<^RcDv;kP@0AYYIN{X9+E36chtP~er0KrR~)c=8Z z3HHB6afOnKnuhig2IFOg+H05aDJd^OQ(q#!tQ~xLA3()Q%_c0XL&N^mp7y5ub-548 z|Imp%sA}ghA0>#&J9vcCGu+_h;^q;%C4T$PT?IuYWffI5-G`6#^bHJ+o;|m)w6cC- z4pHCG z*wNY5-P7CGKQM+GpO{2XO=A|9aLX&JYk$`_h`W3H2Zu+b%_8(^dcf`W~znJ|mV*i^L2Ear~aoKs4tN?AmsYq`K6CooZ^eyv{WUE->WRrA51%}Ks?rpZu zY_q{td)>Y1=+Uqab3$1UF4r^9a-?r%tP_<~lh_ZgoSd*ZtJt-ImWE82IPzqWyvZTK zl}$khqzgcsjefTb4}R1y?s534iX>krO_|RNNCC#ExeLKGtZ7C`+1k0;%19C{mX2|E z#+^CX8ZF^|gFUZ*dmOLB_}O5tf4wK*^_$ibA~dpwqWAGN9dQ-Oc`=zd_yR5ds4O_YXNP(``F=W@VrG#0_6%>XMe;AV=Cn15oHY8U zX%lcC-$2*qoE)T-G&~8+@EM+7cLDgYYvQuI4bNLV>QK*PDdd@~f+)D_tDdUOo0hFM z^h?0Av%@{^Q>r@1=gdu(5mtLUstk$g+O&*c?K^Epihqv{(cRYzDyIDXH2zhts?e*T zaV`3F8|_%K3&ed%hQV$jIQPjOrgf_zbU-OJo9UN8nB$D|=sYD!^+yuVoL}?{okG7* z#zSVM`8U3Kb#Ipc;XmZVrlw}!J|KLx@rw-2z%}iftp%=w`7C%oV!-Ip+O?L$xe{+k z^6pv>ivV~8r>VUZlJr*i4or?ujy9U2mApkt;dOJ{UXdQpY%k!l-e09 zXDXI^3@qnmMw^?7tDa;pd&cv(`Vv8huSo973C;4*a$TvpkLC2q9Y^)>f_~2v}v`H7@&00lad}G@Tga=My z=&ky`dBFWFXuo&!tQO?^S?T3&pKpuaVWemX0}$j#0G6>V-{#rUHvfSgEIYig)^kX- z_*iCm^!zF<@_7HpHoWf^Gp{^5uTur0<>*G`NuO$hNrA zzr1U)E7Vg2_UuJRk*e<8&RIx-n=kt0yy8srs^zKvY^!fdP)EtTB-0B(pXDQTl)d%u zo02>I9{1ZeLi6yp-DlAfB6p_uAi7(*?k}2vYelXLH3(7F=Q#9vfBEwVXUoR*2x1?!nSzOQ+y32>04K0 zd*g)G+}%e|9&*ghgtj%2q-W?v)fqb(*g8*L@^_aDz_k)7@6ez00)Zhq34E2bfX`Dx zUMj-?P2SOS4L@*?offgs=I~p7f1rKIeCdX)FsArcszZ}6{1p!b7iLiaR6 zdvaybwM}uHO~l9OCm7(d&j2&3m>fNy^u`9`t`-EaIITQMJrqJ?Hwy@M&{@ z9y8gGI==w;wz+*kz?&A%ZqSvAgFaYg5M%HKfSwm~-)3WKUScSc?>pru(F3VnT)MEb zfvPRAO+YI3FobAYv!o4p>7pzk&=|WEsEkN%?jw)fF8K;*x0j}SMN75U$$xL^v8&iF z!dpsf%V#qMF@W}|GB49hwFEW+CwE3MbB#b-<}*Q@cpO=g0srYm@@hw~t0kbsC*;Y4 zvP%^S+@%S`Z6`$q^u`^qL}dl9#gwn|e5~3@GnDqZR_Nicac@lpC&JB5lE+#_d&6P{ zWC(~TMeWFw*?^#!rFdmEE{(6^$2!G#?@GvFPdBb zFqHxT4H98){s*E5%Xy`3f4X#_q#f@v)XJL!tKj{;cjHCo?~YK2yb^D)a%pqZ>`eNw zrqOH8bVK1!RL~@Kvmg7~urWA1@Onc7*nzc>u4rj42f2JDBw{8>=BzM84SWH(8p54p z+`*HOg@3wH=T-|30UEI$)c`X4Akp2P<6+uy;$27w5TKsqk$s^-WqZWzZ|p+XWPsi~8l6g6U-P*C&W$XwS=>;ApcieVbA zh@CM1KH5JQ0M{g@mkUAs5#1*o?JwYQ>Hd_nR2DIxhA&4E0zeV2X9y7d8njqUe#Yi! zbSXe6^+C;vZ{@NbnJmJRAKdP#cnc-oo2GiL=+80x>MY@5IcnzuKnt81u{%UM(RqqJ z^&UL|LC!QoHMOV0ce)Watrbu|(Nlr%c*))R9+t0G;+lO@*~`lTtF3YA9o)rxQbkTr zDD9iD&P3Stvxxm*s4mF`h5_GIE=Ja{30R4Q4)#9wMDYa%wGyvu1?AtKrrYi7*E9v3 zf48Dx`|@)P-gAn|*s{D`&oc zJx&cg#m+))xRvW1oHEnotbYwA`aOGTTBJ6BmP>j0_05xS0epO^`&XJEmc&olkmZio zx20yY*Ygl|hmHiWD z*v8chz#buaWJC}o^&c-3JoH5dC2Sr$irGC(9WvF|ZCk9$&Sw8Z8tH}D?4pg~#5Z~N zp%BIK80DEQh2u9J%w22mL9$7e)Bg2;SZnRWG4+ro0H zT*Gd>f8{V*Jt>J>;Fi3F71ykdJWJe=O~rB=93(?hbE$3gbyWiq%w(dpoqzFsCLF{o zRnxB^8tcLpBx35qW}fqtc+Dv4sS}klPCUEw*vORMZ43+Xxk7X&FGIXLlTtJzs2jw? zDDcT2`b2(^@J5pw-UM2l)JVInNEdprWZ}i2TING(8_R3I?^{?{w*n)d{@OaGr~Xos zq1}ZTuWcGK>qG>kc7uOl6q~Ka?xlHb=xx)@>V>R3w%cc~ST)K%J@P`mZc#=V1M^4U zrnbs83SX~j1cIMizkYS~ z+SeslgB~uoCT@L70(j}!Wy_V2AVg&&qNK>nRP#8~iTrfHPc}JK$aju452h;vG^o_D zMN4{PtZ_F5O7?ff^CVr#9zezbiEfKvnJDEzVS^1?E6(&*MN+tXaIjq20cP;O5#uonF21k_L8i? zAIH1g&mmK-%h;=*-3ky`mLS>N9+_IsdgMS81jE0GGp&;$tPpy8Ah9>HZ^LYx5 zukYElWOH)nqo~E8Vz!A^JN(@(`p-seW{qz@KJ3_-%V*bbcmdqG))uOZx(sCqoV_e0 z8J10ZZ0L05+B1?YF`VVPUw+S9Nq7s7?OkwCjZzPwb@FXTrANbG6>}wY?Ar?f!L#^h zG^$&h;=L_StfIBEY@igorT)P&a%8Ai$dy*iGw06J3=MxP7#4=|E%}D`cH}jDR)-Wn zCSL$78cw!KT>4%mI0_dqJufeePd?XQvr|8{53)#I1F}%=U7BVlV*Y z+P$W%^PIBkLDC_y%Q6wInxs6MNSsWh+Dc`HT33Rn6D=kvE}~h9@U@LMmCG5@#*9=I z@jOgEtJk^`_$`O8?pc76&PY4KdBS=eG;%u5Kpx;zmivoz%m{1(^`frWYydHq>?M1;R6 zVdD#PzO(57XdaW0Thyl@d>)Gi1ok!y+HnSY!ejYubpz(Y#!+kNXd7?+i;C?-Ae*h9 zP{a#MpJI%a8M!v0B5;2~^f8-7N^D4u*Br!aa&oM9>;k~vY*Dy@p+-g|jGIf1Bkmat zj{4{gW#={O!yp4X+&JmO#b(XL3axn{VqBqbyxI2-T;cW;M_)>b_{7XM5!-=%NP zF_j0v7N;$2eeUzfS2|y4-#Se5pM&r-uFi4$BQf%R#wJ8)I2mXkT!IVXY~NY#m4CbdxJ=ZEmbiH-qD-XM2F|tsmV&^CkCT1>DE?q#)D@%}+NPn@qeG=bauX zb?*gze|6$Hv(Mr6Wy8`{zIGo!VY9uleS}Qgs`gDA5pam{;-+-{e;~aI8L*a+jIa_h$-iie8x?y#P?(Z6vIH(o*xYIt-{H3NO=j%zU`FG!@?b9W`hPkm1IQ077?vBP% z?xu`8NQ_YX6cD6x(Kf*t3D?bXuSFca{g^Qsjsj&IzucWuKc{mFbSF3(6R&>5Nu1`f z&AQ0G%dpYhcL!Efbl%;Lko@gdzUaH`7J}atYk3&jE~Wii`|961B}`(`()#x1N}a98 zxpA^sGxzKej~uySKxJ0$312pAeIRGoY6~1Kqj=ilVyW^z>{m@|XZO`VB@78|Tugp^ zSjQ597etXI*PNv-$+x3HPis&RV`=kQ(U4aRX4Yt-h^k+*;hMCA?H{_9s-DEJJ)N-% zhM@FPz;>luVy-}C5moadx))%)Z)gr;n+!KP36P1_ezg#Wq=IXTo0IGpWt0wSm}Rmz z7T!N%8i*HU&>s@UUNd~!)+XSIdjWAAugx)+;6Nm9472iQnKj(wt}BgF>CsKyU#o!l zc^+0Kb&uKjX%SjW2|d3$eO;8(!bZ0`&+pO&4FTetE|57O<4|Kjg_PH%ey%6;tN)hRUBv|JqhH#VgPj8XZWjHlz zMwW6eHF4@e?3O$SJIDK909HXEZ*g%E>efd@le{kcLToN0s`*X`aZo1qeQU2qQTpvw z7Y7STfNDDcheE~f?~tT>J`fpANb-vy2ERmRo(+huMb-u2zKN1^vRN8cj60)Xv;68^ zVf-$xbg}831;{)@-UMipRtq+U5srKjwwxmIc8X>NR_GyoQJy6nb7)gHdcyo;{RrVJ zV%9LWyt|V2F+L!ymo}+5n8kH@Lh2b-J4r)I(*2zp|3q<`C-g8u4)1?s_vroSt7eF& zov$KxH;~@-@7mxa9tItIFLEe zgl522>lV7m(AbzyrCRT54+#V6g>K#Um10N9GUeSsK3N0)!3Q1-si-9B$mmd0PntSyXOy+e&O? zp-Yj#4JXsWt%3t&J{km)Xj%x?fGvHB3o>?I0+X7{Xm()&>i!j|TThD8tUcqgSKU}d z>gWau%e26WhmEO)blRYk-N0DqxPzWwla=rU-A+q}9z}$Bf2rTnO8XvoQ3|Gf_%r8hI zv=T`!L=LYDKx9FCh|rh=xJ*LT5z!Fqe)nb1KhrmR=V>XS+VkQhpl@I9by9&hHjiXX zzNJ8V{!Ur){Z5vi90r@lOV@T>9mwY>vcqcAxYnIxr_J@q3+mJEs6}1=u(Y1BZ>G(` z5dR^U_5rm|tKeB)2w#v1sY&w5`n$S}%N#m2_dW`^Os?857O{ylW%KXT*WTdkXPWw<83Fh>}o* zV(sgsv9aGa>$&2|j#AfI3*GKNt~BjPrr5humhkbjf9ivg!|sf(fUUYB&qexjtLO?eApcc0-vGLR2|LVqNKoNj(3&b(m_uZkwQ=*Ff@KL;PEZFUw09@QCrg zoiW%O|L5?h>a)%tGWdfrjC7N7RR&R(?7I{iE9duB_nkt+tYqEqGNGAY{$T_;-_2PM zQw#qAK=xX!R~e9tO%bkyF^RaZ7 z<;YkNH7s+5a{Xn1mLFR>fO^tyt>#?=hGO@7x+$7FSi~u5yB2P22FVk0W$4x~0E`vV ziVmW8D*D<6w)OASeo4FAeVVlIYx5yVZlx}3?mh%7P7qp6Q4>~z`8>K?MJ^zg3q zOZ_rAP|s}O{YqTAUHa#czUcUzvo6EtgkR-Nj_K>l#&eK#=>;GubU~1s#}9(&J{*ZnC6~X3=LJ3 z8oz(#bM<=1Q^spX_asN6l7x~|WSgIzvc1~{qdJSp_t9|Rv9Q16Zh7U>yxdSi|239= z$M{9PRU!LKKfE?cEGhO}%*K#5UfX4c7j!S>&wAEx<#QiBIdgrTZ1xF*cA+&%(Tz52 z&O1c>IJ~ZG$7W|29IUJ{aFF$~+RUuGc>e+rH8a+me5s*L6AE6WNJ0-t;4B~CD>v0% zZimOH*J7ho5Qvl|>kdBC(KMlKnc2H;=NV;WbCSdoZ2G7(j}AvFS@iSf=gW6NAMwB+ zgsT63BKDc)B|EEw4*%1rrGU<|&~|DK%|a4Om-aO;Xq*~4cH1lN&aIkXzZz;#xDUMlSyzN_0+F!|7MpS~WO+`mDm zs(q&jL~B%e%Tyw5PTAG=b*e3{I#9FLXB1;^<9Y_Y%;KReLiZ+nVSf3&-D;M#7XZ40 z;i9Vr5<*#q<*q4@U7CYuzLqa^mzNX&Qr=GB1xTwo7OdCtS%hxS30|&d*_r(aM7>2xWtT~(ay+O;D!qK>B)AX&xRQhm!(zs>5bi^*J zC2e0W^0w(n3FHF6j~v2GjxQ=0oA5vky@wz+8&*#D>dxtBa=JpAPtH8?+Dzx-Q`2B2 z38%&W*0ENuV2plz ziml@J3LbocNvzlStgl`GwCZfV$I@u;#y$m=7$etP2TN6K!R;qesY`bZPrWm+6ur zi6J%%a~aOZ$2PsAWy&2aGKVWxrT(lpRA=g`yx(e5(U6%Fym~1z6r3M}p8j|npqFhP z2u3mgig)}SUvpbxr_&v+D)5oq<;kNpYIL}c7_<$*Lp04be=f>CNs_2V!qo97qOs#j z9{XDX`LSj(r{Z`0*YOY$xbQ9OInU3W#Rh^&_KEG6>O~)-WP5DvQXIiGk{&WG$Y3Dy zbK8jT)Ek>T6ZD+0g^t2-duwWnCT=VWg=P9n`ipC2A8K|7GhziJ12|)aYa7QkbeMeU z*(9b8UM%+>NG`ng2IIUtl2+H}rG%Cjc1 z5(5pI^jZny1}{Gwv^+3aOM{$g98h?^jfC8lLOG&wgTFTTd&XA%Z&4 zWu&}eu_IJPTh<(~mYWkycn8laE`Us2<|=aIK;2xsF5A6k_=9t}wrXWDK_kXT@t_NZslHfMtiitpFJllX@UN=*5t*xWy`o z?t*Vm+T#NsWC|GeiG8C^G~Cd?uX6%CoFeHG8PJFEmospM25YcZUa3lhq2++Snaryv z(+m&JCiu;s_Ae$ytP4;OKO%0~O0-W_R{!&y?PZs5u5A3P!@mY(6_MK$Pg5X=`>4tm zNN~kp7h3Vxi)9;#3&5v2m-F}MQO$~{5VATken>gdi=xCXr=KdldX>K->6<5cu+7eN z70b8Ht#AOF*2SW`>7+HFd?)AM&I&y21|T$>Y=5Whg+3cRc3vHkWD!|qjy+z*EwOCk zJaN1i0FJpJ)^X^T>%zVDgwnLc@z*OBAOA_BVoK>%(O^#%)rMb2UqQ;whh{WEcj{4n zO;S)F#rs>=a3sciu5zk;Yq`yP zGlN=9oL>(4F%5Qq!jaxtdF|gI8J~FMbR+DOCAoD){FdSCS1jr=+@d$cg=(xuLhBN% zu!7;aLWJKTs)SoPIEkY%$8glM@pz>%TVw9G9nJQ#vXT^c)o3V1575vV%-UmO0LTxy zIF)7JsH$k0JKkq6Z0Mrx!{`yWyT-)_0ZpDOry>Y{PgDZGYD$!Rwe&Mm!J*%2OmH*|hQwDBSk>`PF-4Q)XfsN0T% zWP6t8ExE96{&v^EzW&Xdxkjbq=T#v}r7-4?TtdMRLJbbI=|T(%b~rsOoq#ky^A?}( zjanvhq_sx@N*5Xp^DSg$A=7$HC14t=I~FIpT?>R~W)T+v%5xrT^6j&z)@yH1-qyw> z2>o?!&ZkRXzxPnl;(}NLA5JI`air~4{(a(_y<6omC+qSud%3OCo>+tJn z=g$kD6}IvL$(I?((Ob2RV%z%3%!i|*i#GvIp>ZoA$2bvRowf7JghzNBnS!XCsJM=N z#}b*|_$fxT76^>-LNfOBt;+u5_#C!4#Z02w+BRm9gQ^k5yJe>J?~{lH_)rS5an8;m zcgPRCmS9WmbH7#AoO2uM=VFOmKYKL&h8$xLziw20E5wOb){g!yuWZCuCaCvT&FG9p zG@e%=Mr?iOv}-(PG;~f))^q7H4?;iXD%Gd>UjTdv@@hR4R7)T^voCkrdT_<-PNOCt;+{dwHqwz8CC-%Ga6 z4U87*RfUxS)q&zqly6##H)V>?-EY3p@-co^NwF>c2rMEM6NCf3p>;$wqG6 z2l=B&;3eg5mXBFuU@b{Be6Z=OXseG;(Kj^!V4V?<(K%!a(o+JiOYoYa8K)nQ%I`Q01JNfxav=-Po(3ZPSLcbX))u0)7$|bGnDpn-0s$4B$hWIfWD-Ty7{n=!~MJ zl!_KvvJuNM`}3dI%( z!js}cm`M1}a}ARB5->q-7i}1rAb^qMaE_n;y{Cr_w?2+y$xzX1P9c-B9~u4nc&B%% z5e>6d9Gl#z4Q}*aEMd~{-!LM#s2?EG`yrJXV||CylA`tDy4RgD^+ew|$j=b*FW2I{ zOk6Rv>6opp!^VX2!fkHKq|T(SKxdHoWNk3*^9GbjQ=eFJr)XT&7s~ioVfrBkBYYqF zel+XQP40iKc#nRSD4${_Nk_GuGaG-n_H%tWc;Mv#0|;%lfhH%VyZ^lL4K$gT@v|Q| zU0Llgb|B`Eg8f~6>z>L8Xm8;=Ec^T}3G{?;0Z5cq$g6n}r+!>#)JgdU*S4Xj1bgg# z2+OZ1nfw^eSRP%VBA?dDp4UvV-n&OO`x4|e_8mseh+Y#k(j*^r?+U22Jsc4!x3&l)QsYvw*iVp zr^fE3_a)c9Kur2Q3!z=@2ewOv4Jbk|LTp%Pst8a@eRY6L6KCyMcIWC|dU%H*5C;ko>W@_ene6(&? z5DF>4qm_v3=-nCPn|IQm-+a{1-)rX)nAeL;`Wwnf_VLvxy@Y~;5~{*l8xFKb zL$bF?u;J}vqMuwkU3vZcM6``M zNDV*Mc{o3Ac~!7KxpDFpxhL&^4ZraNnYxuzV?=0sYSQN&$OG={SXXz}-*(7tb z_ybqfGtu{b8u^A!7TUMy+Emniw#pyfUX;DQrmfwrx#PNlfeyVY7>Vf|xO+4TUd}r% KIyx4-nEoF^-Ix0S literal 0 HcmV?d00001 diff --git a/nst/Models.py b/nst/Models.py index 1329423..6b2eab8 100644 --- a/nst/Models.py +++ b/nst/Models.py @@ -296,7 +296,7 @@ if __name__ == "__main__": # model = models.vgg19(pretrained=True) ph = "weights/vgg19-dcbb9e9d.pth" - transfer = Transfer("./ori/2.jpg", "./art/2.jpg", ph) + transfer = Transfer("ori/2.jpg", "art/2.jpg", ph) t = time.time() dt, img = transfer.fit()