diff --git a/federated_learning/yolov8_fed.py b/federated_learning/yolov8_fed.py
index ae2267d..baf3c76 100644
--- a/federated_learning/yolov8_fed.py
+++ b/federated_learning/yolov8_fed.py
@@ -69,7 +69,7 @@ def federated_avg(global_model, client_weights):
 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 = YOLO("../yolov8n.pt").to(device)
     # 设置类别数
     global_model.model.nc = 1
     
diff --git a/image_fusion/yolov8n.pt b/image_fusion/yolov8n.pt
deleted file mode 100644
index 0db4ca4..0000000
Binary files a/image_fusion/yolov8n.pt and /dev/null differ
diff --git a/federated_learning/yolov8n.pt b/yolov8n.pt
similarity index 100%
rename from federated_learning/yolov8n.pt
rename to yolov8n.pt