Anchor-x/Graduation-Project
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

文件结构调整

Browse Source
This commit is contained in:
Yunhao Meng 2025-05-05 17:03:41 +08:00
parent ba4508507b
commit 310131d876
3 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
federated_learning/yolov8_fed.py
View File

@ -69,7 +69,7 @@ def federated_avg(global_model, client_weights):
def federated_train(num_rounds, clients_data): def federated_train(num_rounds, clients_data):
# 初始化全局模型 # 初始化全局模型
device = torch.device("cuda" if torch.cuda.is_available() else "cpu") 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 global_model.model.nc = 1

BIN
image_fusion/yolov8n.pt
View File

Binary file not shown.

0
federated_learning/yolov8n.pt → yolov8n.pt
View File