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 f2990a5..0000000
Binary files a/federated_learning/test_data/client1/train/images/img1.jpg and /dev/null differ
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 1c8ea8f..0000000
Binary files a/federated_learning/test_data/client1/train/images/img2.jpg and /dev/null differ
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 dfb4988..0000000
Binary files a/federated_learning/test_data/client1/train/labels.cache and /dev/null differ
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 f2990a5..0000000
Binary files a/federated_learning/test_data/client1/val/images/img1.jpg and /dev/null differ
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 1c8ea8f..0000000
Binary files a/federated_learning/test_data/client1/val/images/img2.jpg and /dev/null differ
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 ac3aea5..0000000
Binary files a/federated_learning/test_data/client1/val/labels.cache and /dev/null differ
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 f2990a5..0000000
Binary files a/federated_learning/test_data/client2/train/images/img1.jpg and /dev/null differ
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 1c8ea8f..0000000
Binary files a/federated_learning/test_data/client2/train/images/img2.jpg and /dev/null differ
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 1b56cb1..0000000
Binary files a/federated_learning/test_data/client2/train/labels.cache and /dev/null differ
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 f2990a5..0000000
Binary files a/federated_learning/test_data/client2/val/images/img1.jpg and /dev/null differ
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 1c8ea8f..0000000
Binary files a/federated_learning/test_data/client2/val/images/img2.jpg and /dev/null differ
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 c20be88..0000000
Binary files a/federated_learning/test_data/client2/val/labels.cache and /dev/null differ
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  # 关闭冗余输出
             )