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_Cq2t&#4Pk9H-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  # 关闭冗余输出
             )