Compare commits
2 Commits
8cd6df4527
...
9d99b00e55
Author | SHA1 | Date | |
---|---|---|---|
9d99b00e55 | |||
dd0e0d869c |
.gitignore
dataset/train2/images
federated_learning
1
.gitignore
vendored
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.
@@ -1 +0,0 @@
|
||||
0 0.5 0.5 0.2 0.2
|
@@ -1 +0,0 @@
|
||||
1 0.3 0.3 0.4 0.4
|
Binary file not shown.
Before ![]() (image error) Size: 225 B |
Binary file not shown.
Before ![]() (image error) Size: 225 B |
Binary file not shown.
@@ -1 +0,0 @@
|
||||
0 0.5 0.5 0.2 0.2
|
@@ -1 +0,0 @@
|
||||
1 0.3 0.3 0.4 0.4
|
Binary file not shown.
Before ![]() (image error) Size: 225 B |
Binary file not shown.
Before ![]() (image error) Size: 225 B |
Binary file not shown.
@@ -1 +0,0 @@
|
||||
0 0.5 0.5 0.2 0.2
|
@@ -1 +0,0 @@
|
||||
1 0.3 0.3 0.4 0.4
|
Binary file not shown.
Before ![]() (image error) Size: 225 B |
Binary file not shown.
Before ![]() (image error) Size: 225 B |
Binary file not shown.
@@ -1 +0,0 @@
|
||||
0 0.5 0.5 0.2 0.2
|
@@ -1 +0,0 @@
|
||||
1 0.3 0.3 0.4 0.4
|
@@ -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 # 关闭冗余输出
|
||||
)
|
||||
|
||||
|
Reference in New Issue
Block a user