Compare commits

...

2 Commits

Author SHA1 Message Date
myh
9d99b00e55 更改最小测试示例 2025-04-21 23:50:41 +08:00
myh
dd0e0d869c 忽略缓存文件 2025-04-21 23:50:12 +08:00
23 changed files with 3 additions and 10 deletions

1
.gitignore vendored

@@ -300,3 +300,4 @@ Temporary Items
# project files
/whl_packages/
/federated_learning/runs/detect/*
*.cache

Before

(image error) Size: 1.5 MiB

After

(image error) Size: 1.5 MiB

Binary file not shown.

Before

(image error) Size: 225 B

Binary file not shown.

Before

(image error) Size: 225 B

Binary file not shown.

Before

(image error) Size: 225 B

Binary file not shown.

Before

(image error) Size: 225 B

Binary file not shown.

Before

(image error) Size: 225 B

Binary file not shown.

Before

(image error) Size: 225 B

Binary file not shown.

Before

(image error) Size: 225 B

Binary file not shown.

Before

(image error) Size: 225 B

@@ -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 # 关闭冗余输出
)