From 9d99b00e550e95bda9b08ba30a19b27d25603cde Mon Sep 17 00:00:00 2001 From: myh <yunhao.meng@outlook.com> Date: Mon, 21 Apr 2025 23:50:41 +0800 Subject: [PATCH] =?UTF-8?q?=E6=9B=B4=E6=94=B9=E6=9C=80=E5=B0=8F=E6=B5=8B?= =?UTF-8?q?=E8=AF=95=E7=A4=BA=E4=BE=8B?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- dataset/train2/images/{02.png => 02.jpg} | Bin .../test_data/client1/train/images/img1.jpg | Bin 225 -> 0 bytes .../test_data/client1/train/images/img2.jpg | Bin 225 -> 0 bytes .../test_data/client1/train/labels.cache | Bin 920 -> 0 bytes .../test_data/client1/train/labels/img1.txt | 1 - .../test_data/client1/train/labels/img2.txt | 1 - .../test_data/client1/val/images/img1.jpg | Bin 225 -> 0 bytes .../test_data/client1/val/images/img2.jpg | Bin 225 -> 0 bytes .../test_data/client1/val/labels.cache | Bin 916 -> 0 bytes .../test_data/client1/val/labels/img1.txt | 1 - .../test_data/client1/val/labels/img2.txt | 1 - .../test_data/client2/train/images/img1.jpg | Bin 225 -> 0 bytes .../test_data/client2/train/images/img2.jpg | Bin 225 -> 0 bytes .../test_data/client2/train/labels.cache | Bin 920 -> 0 bytes .../test_data/client2/train/labels/img1.txt | 1 - .../test_data/client2/train/labels/img2.txt | 1 - .../test_data/client2/val/images/img1.jpg | Bin 225 -> 0 bytes .../test_data/client2/val/images/img2.jpg | Bin 225 -> 0 bytes .../test_data/client2/val/labels.cache | Bin 916 -> 0 bytes .../test_data/client2/val/labels/img1.txt | 1 - .../test_data/client2/val/labels/img2.txt | 1 - federated_learning/yolov8_fed.py | 4 ++-- 22 files changed, 2 insertions(+), 10 deletions(-) rename dataset/train2/images/{02.png => 02.jpg} (100%) delete mode 100644 federated_learning/test_data/client1/train/images/img1.jpg delete mode 100644 federated_learning/test_data/client1/train/images/img2.jpg delete mode 100644 federated_learning/test_data/client1/train/labels.cache delete mode 100644 federated_learning/test_data/client1/train/labels/img1.txt delete mode 100644 federated_learning/test_data/client1/train/labels/img2.txt delete mode 100644 federated_learning/test_data/client1/val/images/img1.jpg delete mode 100644 federated_learning/test_data/client1/val/images/img2.jpg delete mode 100644 federated_learning/test_data/client1/val/labels.cache delete mode 100644 federated_learning/test_data/client1/val/labels/img1.txt delete mode 100644 federated_learning/test_data/client1/val/labels/img2.txt delete mode 100644 federated_learning/test_data/client2/train/images/img1.jpg delete mode 100644 federated_learning/test_data/client2/train/images/img2.jpg delete mode 100644 federated_learning/test_data/client2/train/labels.cache delete mode 100644 federated_learning/test_data/client2/train/labels/img1.txt delete mode 100644 federated_learning/test_data/client2/train/labels/img2.txt delete mode 100644 federated_learning/test_data/client2/val/images/img1.jpg delete mode 100644 federated_learning/test_data/client2/val/images/img2.jpg delete mode 100644 federated_learning/test_data/client2/val/labels.cache delete mode 100644 federated_learning/test_data/client2/val/labels/img1.txt delete mode 100644 federated_learning/test_data/client2/val/labels/img2.txt diff --git a/dataset/train2/images/02.png b/dataset/train2/images/02.jpg similarity index 100% rename from dataset/train2/images/02.png rename to dataset/train2/images/02.jpg diff --git a/federated_learning/test_data/client1/train/images/img1.jpg b/federated_learning/test_data/client1/train/images/img1.jpg deleted file mode 100644 index f2990a520b59902985c56bafc9caecae2d58fc20..0000000000000000000000000000000000000000 GIT binary patch literal 0 HcmV?d00001 literal 225 zcmb7;%?-jZ427R*iDD;hofhWl0jUy9!6==FDL60$KbLOeCY<Pz?@88swq)C$wm>h_ zc>)nYS+EaIsCI=?t4!_cf_HB3*o&}dme@rJZD4Lv({`#!CH6zFL-uNS1dkarQGz{i b{O-puoZAP@2e;T8zWg%ccHK_?PNTh6b5a#b diff --git a/federated_learning/test_data/client1/train/images/img2.jpg b/federated_learning/test_data/client1/train/images/img2.jpg deleted file mode 100644 index 1c8ea8fd156dca28b8a89aae65b15a70253b1f01..0000000000000000000000000000000000000000 GIT binary patch literal 0 HcmV?d00001 literal 225 zcmb7;OAf*?3`EDYM6r{$P7CKL5>h2NRWHGPSa1kFo8H7tSkWbKBx}Z&Y}?Zo=ykeG zAOa{0_Q4s|u25=~sa;+0&g~t05%$ayyC|Uz%uQ<APBp2-e&}_`UhR(HF=Hl5um_Ic a{rH7*`=I&Y7JI{&Uq;+c+sWT)wATt{UKJGp diff --git a/federated_learning/test_data/client1/train/labels.cache b/federated_learning/test_data/client1/train/labels.cache deleted file mode 100644 index dfb49882bdae086997c5aa51114b5b32ecdeb63e..0000000000000000000000000000000000000000 GIT binary patch literal 0 HcmV?d00001 literal 920 zcmb`FU279T6o&KFMq351ig*=oObTtxHXn^eTA`@mY8nJDgp_4w=WJ%%{RlIY+Gq>8 zv9>UGIbPJ?)cz)&*(iwAD?P*6*)w})X5WW*)_eD+|305P%6+ObkiMvHRjMa@)wN1B zNQDw4@lwISF25j=1hXzV8G)&5t810h%D=_ZwDk3EF_(p2NyhPLQuq8+Ks_EuDkMUX z3F<qZ06$Hn5@TQC4D~|RiMpJydCLsl+-Ka)Q~ZK&vF7Gic~LJgH5q~JFYI;k;(Y$8 zsPiY->-AoJH<3Nu9fkB_A5XES7b8L;lDLmGTb9Gv3&IFc_n&VKUJAm-M1^Ux-WTZs z_-YV<fkPB9F9ITxFc}UMNaZo2$iR<6NR%@$7{g=`#$*W6Ohc!BFd7;a8A)`f?(__G z(Kpxf0$;F7c=@LeH}B@pkT2^((E44%(?2BStc2Z?1&QWWFH)Kw!_5bl%z+OtcJG<q z9h=uzueWVx8*fP%#^$dr0)6`sCZja8RZlM^sfbAweguZ+df8s!8IQz7p)MUyK5!fx zWPIhnqT0w%@%wW~{K4M<89he)bjNDok8bMvrTW|D<+jc1tE()t9qaqZANP3UoI8AD z47$XL<fxwkYf=E9j)_Ll=@8azG@b4yBW=>6w9BXi?T$}^PSEwq2Jt&jT7gf=W|z>Q q&4{t60BNi&jVTLmE*k|4ESAFzarp>@G>$TMaO#bE6OF-;)_(wTAV$Oh diff --git a/federated_learning/test_data/client1/train/labels/img1.txt b/federated_learning/test_data/client1/train/labels/img1.txt deleted file mode 100644 index 2aff1e2..0000000 --- a/federated_learning/test_data/client1/train/labels/img1.txt +++ /dev/null @@ -1 +0,0 @@ -0 0.5 0.5 0.2 0.2 diff --git a/federated_learning/test_data/client1/train/labels/img2.txt b/federated_learning/test_data/client1/train/labels/img2.txt deleted file mode 100644 index e82ecae..0000000 --- a/federated_learning/test_data/client1/train/labels/img2.txt +++ /dev/null @@ -1 +0,0 @@ -1 0.3 0.3 0.4 0.4 diff --git a/federated_learning/test_data/client1/val/images/img1.jpg b/federated_learning/test_data/client1/val/images/img1.jpg deleted file mode 100644 index f2990a520b59902985c56bafc9caecae2d58fc20..0000000000000000000000000000000000000000 GIT binary patch literal 0 HcmV?d00001 literal 225 zcmb7;%?-jZ427R*iDD;hofhWl0jUy9!6==FDL60$KbLOeCY<Pz?@88swq)C$wm>h_ zc>)nYS+EaIsCI=?t4!_cf_HB3*o&}dme@rJZD4Lv({`#!CH6zFL-uNS1dkarQGz{i b{O-puoZAP@2e;T8zWg%ccHK_?PNTh6b5a#b diff --git a/federated_learning/test_data/client1/val/images/img2.jpg b/federated_learning/test_data/client1/val/images/img2.jpg deleted file mode 100644 index 1c8ea8fd156dca28b8a89aae65b15a70253b1f01..0000000000000000000000000000000000000000 GIT binary patch literal 0 HcmV?d00001 literal 225 zcmb7;OAf*?3`EDYM6r{$P7CKL5>h2NRWHGPSa1kFo8H7tSkWbKBx}Z&Y}?Zo=ykeG zAOa{0_Q4s|u25=~sa;+0&g~t05%$ayyC|Uz%uQ<APBp2-e&}_`UhR(HF=Hl5um_Ic a{rH7*`=I&Y7JI{&Uq;+c+sWT)wATt{UKJGp diff --git a/federated_learning/test_data/client1/val/labels.cache b/federated_learning/test_data/client1/val/labels.cache deleted file mode 100644 index ac3aea58d5db30bddfce0ce7529d5961b2f700f9..0000000000000000000000000000000000000000 GIT binary patch literal 0 HcmV?d00001 literal 916 zcmb`FO>fgc5Qg*BhPE6yRfto#kW^}e-8vtxP>WOn5(c+I;((OOTHBMXi$9{>O=zH% za-pQs-ZC7@Z%Tg??7AQ!R8BCf@p{&?yM7<NbNB7*qj$x^xbUeS0uNYyw^lzptZ&xp z{fr4lQZHj6=;h}m=AhO&B|}hkV`H;+Ui-ILomRfClnZ&t^>maBCrvNN7&MbnEF!`f znV`JuF$l7h3pNS_&QUJqohYkmc+)aRr|=mMiavh9H`s8B8?-E!LNOVF?k^p7@p5+a zsVs|U=(_I9?<(?#voFxUJjQcu$mN*$5OX}nhAyj7;`O5#P`b}{doLIXM?^$fx^={| z69`1F4+;(u5PC5XmPYBICx8ntB!cvUID%A|y)lV<Q9=g5)if}hC&Pi#kaI+5>P#<C zmIHMuEzu>dgI0g4aEeax0_mD8^{w9pJo!UFK?^vjXpE>{<+AT*r*Qj$6?Ndl%Y%EW zcVFlA;?<tceB&JsgGBwa1|V17!(^C6x^m@enz4k$(MJezCfD=@o^nV+1j@?k<O9W# z0!G*WBdF~h6up0gh(7om9;HS}pYCe~^wDiyzchcly4uruy;$U#@7Ov<`q-t$jGA<N z1hPU2rzoGcNy{>bZNWD9L8oOC(;}@F7-7fg+JV*bEwgPH0d&o7-|YJx@Y{A^+ufa3 t(6MZPN7<7BS2pI#kR_*(kCMif@Iel^ItIp-o0J*MrqSF%WiIraKL8u^Lcjn3 diff --git a/federated_learning/test_data/client1/val/labels/img1.txt b/federated_learning/test_data/client1/val/labels/img1.txt deleted file mode 100644 index 2aff1e2..0000000 --- a/federated_learning/test_data/client1/val/labels/img1.txt +++ /dev/null @@ -1 +0,0 @@ -0 0.5 0.5 0.2 0.2 diff --git a/federated_learning/test_data/client1/val/labels/img2.txt b/federated_learning/test_data/client1/val/labels/img2.txt deleted file mode 100644 index e82ecae..0000000 --- a/federated_learning/test_data/client1/val/labels/img2.txt +++ /dev/null @@ -1 +0,0 @@ -1 0.3 0.3 0.4 0.4 diff --git a/federated_learning/test_data/client2/train/images/img1.jpg b/federated_learning/test_data/client2/train/images/img1.jpg deleted file mode 100644 index f2990a520b59902985c56bafc9caecae2d58fc20..0000000000000000000000000000000000000000 GIT binary patch literal 0 HcmV?d00001 literal 225 zcmb7;%?-jZ427R*iDD;hofhWl0jUy9!6==FDL60$KbLOeCY<Pz?@88swq)C$wm>h_ zc>)nYS+EaIsCI=?t4!_cf_HB3*o&}dme@rJZD4Lv({`#!CH6zFL-uNS1dkarQGz{i b{O-puoZAP@2e;T8zWg%ccHK_?PNTh6b5a#b diff --git a/federated_learning/test_data/client2/train/images/img2.jpg b/federated_learning/test_data/client2/train/images/img2.jpg deleted file mode 100644 index 1c8ea8fd156dca28b8a89aae65b15a70253b1f01..0000000000000000000000000000000000000000 GIT binary patch literal 0 HcmV?d00001 literal 225 zcmb7;OAf*?3`EDYM6r{$P7CKL5>h2NRWHGPSa1kFo8H7tSkWbKBx}Z&Y}?Zo=ykeG zAOa{0_Q4s|u25=~sa;+0&g~t05%$ayyC|Uz%uQ<APBp2-e&}_`UhR(HF=Hl5um_Ic a{rH7*`=I&Y7JI{&Uq;+c+sWT)wATt{UKJGp diff --git a/federated_learning/test_data/client2/train/labels.cache b/federated_learning/test_data/client2/train/labels.cache deleted file mode 100644 index 1b56cb1e9da40278d816a908f5e81178d2dfa26b..0000000000000000000000000000000000000000 GIT binary patch literal 0 HcmV?d00001 literal 920 zcmb`FU279T6o&KFMq351ig*=oObTtx=4;a;jSy6DH7$Y{Ldr6`dp5ImKf=tUHrhgN ztO?9rju-VerN2pMHVPv3O3!e1_ROA{+4tdH^xwWddY8?dW<HfY;4Ulgm&)gd<(*R5 zPncjNwiD)oUVctO4r-lKG6Gdsws%SwrGJZ!Y2oYLd?pRK6_2CQq-whfgK9JmML-xM z6O?ys25u5_!N#t@1<JLw6J;^>RxJy(GM}-Rb?^(m!HSjLrg^#MiOC3bf9|k@^V#ZC zUS`j+-|xTtt|EO{U4hR07%#9Q*F)k!$nh8}x-15f?FS*CJbJc2c)^G_CL&1U-6NKq zfGY+*C^$rbXNN#o9K^$c04{8g2r_WP0Af)eD2zco2qH2BuBM?;JsAy^ij*W;Q)_yO zGViKuX^yUGCA9cchn2Ommq<5d&TsxM;mIEoGFn2fph2Q~mFtd^oWacp7Sw?c=e>KX zx2yAd`RYJty73l=VWj?ABapY>!(@~My6Ve~IAIY9gOA|hOm6B6Y~_)N2$Y4h$p?yK zg^X_fS5);B6}`WLL?8SOkkX^XPrF(JeRNaTFV)|!t`2lwFPCYiJ9dweKJL@nj2g5) z23eqlQ<P6l(lCkPc1(ZIsX=|O?KB!jjntf$=@HXuwZZe7drfHjuHhQ3hT+sZq|-5* tW*b_L*HRW`z?GG`GG)%nq$8(+MSPedE}jD8%2CP=MzvOLpfVUb)gRS6MAZNQ diff --git a/federated_learning/test_data/client2/train/labels/img1.txt b/federated_learning/test_data/client2/train/labels/img1.txt deleted file mode 100644 index 2aff1e2..0000000 --- a/federated_learning/test_data/client2/train/labels/img1.txt +++ /dev/null @@ -1 +0,0 @@ -0 0.5 0.5 0.2 0.2 diff --git a/federated_learning/test_data/client2/train/labels/img2.txt b/federated_learning/test_data/client2/train/labels/img2.txt deleted file mode 100644 index e82ecae..0000000 --- a/federated_learning/test_data/client2/train/labels/img2.txt +++ /dev/null @@ -1 +0,0 @@ -1 0.3 0.3 0.4 0.4 diff --git a/federated_learning/test_data/client2/val/images/img1.jpg b/federated_learning/test_data/client2/val/images/img1.jpg deleted file mode 100644 index f2990a520b59902985c56bafc9caecae2d58fc20..0000000000000000000000000000000000000000 GIT binary patch literal 0 HcmV?d00001 literal 225 zcmb7;%?-jZ427R*iDD;hofhWl0jUy9!6==FDL60$KbLOeCY<Pz?@88swq)C$wm>h_ zc>)nYS+EaIsCI=?t4!_cf_HB3*o&}dme@rJZD4Lv({`#!CH6zFL-uNS1dkarQGz{i b{O-puoZAP@2e;T8zWg%ccHK_?PNTh6b5a#b diff --git a/federated_learning/test_data/client2/val/images/img2.jpg b/federated_learning/test_data/client2/val/images/img2.jpg deleted file mode 100644 index 1c8ea8fd156dca28b8a89aae65b15a70253b1f01..0000000000000000000000000000000000000000 GIT binary patch literal 0 HcmV?d00001 literal 225 zcmb7;OAf*?3`EDYM6r{$P7CKL5>h2NRWHGPSa1kFo8H7tSkWbKBx}Z&Y}?Zo=ykeG zAOa{0_Q4s|u25=~sa;+0&g~t05%$ayyC|Uz%uQ<APBp2-e&}_`UhR(HF=Hl5um_Ic a{rH7*`=I&Y7JI{&Uq;+c+sWT)wATt{UKJGp diff --git a/federated_learning/test_data/client2/val/labels.cache b/federated_learning/test_data/client2/val/labels.cache deleted file mode 100644 index c20be8895ac7297aa9de26ed783cc79bd2cc989b..0000000000000000000000000000000000000000 GIT binary patch literal 0 HcmV?d00001 literal 916 zcmb`F&rj4q6vy`$tbhkk#CTdS?vlVtTgooPfC*6(tA&_&Fj>+}JFh#%{@BdSBCC+> zMTBH-JP-Ic!QZ4aMH6H2q;K*%ebYBH{eJSj@ZY}PdsnR-Rz9^Uh={j0TkYfB_HwHo z7hH0Z2L-2KmY<VUfL<4jj6m0&rRCO1>)&F2+W0zGuauz{^Kmwsbc3ki(9OoFObF*> zg6eL-Au4hq`8bkzfoi7gMAgjcP0Iy(mCv|S4e<-U!H!p5Vs$k`<zxh=zqZ@Q^Rt^z zbyYn^-}hgB*HJ#aZHeLeKAvDl&88%TRNy{#OxaAbAWl+1<v-gTyx@e6iA;)oWsesJ z5XnId8V->_gA@qQlYBUkAY?#^B!ehTAeZ*wkfehoBSR2+8d}|h(NJqBIiff9rl+Xt zk-n7G*o@J^nm<)|Rj+!AY(drHwciCi`9na(2-s;DjHqALY#0_taQlG`ec;3MoqM`> z+vN4tt1Xk|#ybLrnf_-3K+U~}$*4$7<*WI;;2BAik3jKEEtm@e?U0N}RE?v_2Zmz} zj4l30P<9E5-M>M^9{dfDR->g)w~Ye!=(et3n!jCMZkfEkx+=5Wv9gcsvCme|n8oZd zs0Jf~p?d1{)=6kNHtDa&u4}n&+z0BcSuS<0-UeL{dp**(Ahf_Cq2tPk9H-k>(! ra9jwrJvj(%W1$UM^D5=28C;ndmVldwz=d{`HiOk&?K-H<g<<yxCTc?u diff --git a/federated_learning/test_data/client2/val/labels/img1.txt b/federated_learning/test_data/client2/val/labels/img1.txt deleted file mode 100644 index 2aff1e2..0000000 --- a/federated_learning/test_data/client2/val/labels/img1.txt +++ /dev/null @@ -1 +0,0 @@ -0 0.5 0.5 0.2 0.2 diff --git a/federated_learning/test_data/client2/val/labels/img2.txt b/federated_learning/test_data/client2/val/labels/img2.txt deleted file mode 100644 index e82ecae..0000000 --- a/federated_learning/test_data/client2/val/labels/img2.txt +++ /dev/null @@ -1 +0,0 @@ -1 0.3 0.3 0.4 0.4 diff --git a/federated_learning/yolov8_fed.py b/federated_learning/yolov8_fed.py index 95a56f8..8b82916 100644 --- a/federated_learning/yolov8_fed.py +++ b/federated_learning/yolov8_fed.py @@ -71,7 +71,7 @@ def federated_train(num_rounds, clients_data): device = torch.device("cuda" if torch.cuda.is_available() else "cpu") global_model = YOLO("yolov8n.pt").to(device) # 设置类别数 - global_model.model.nc = 2 + # global_model.model.nc = 2 for _ in range(num_rounds): client_weights = [] @@ -96,7 +96,7 @@ def federated_train(num_rounds, clients_data): local_model.train( data=data_path, epochs=1, # 每轮本地训练1个epoch - imgsz=128, # 图像大小 + imgsz=640, # 图像大小 verbose=False # 关闭冗余输出 )