Commit Graph

21 Commits

Author SHA1 Message Date
1822aca36b 优化验证集文件读取逻辑 2025-10-23 13:06:38 +08:00
ac4af34802 暂时禁用缓存加载功能,移除缓存文件以避免潜在问题 2025-10-23 13:06:24 +08:00
52382e460d 移除训练过程中的调试打印语句 2025-10-23 13:06:13 +08:00
TY1667
76d7149512 更新.gitignore,新增忽略结果、日志和权重文件;更新README.md,添加参考链接 2025-10-19 21:31:18 +08:00
TY1667
f23a22632f 新增testcode.py文件,包含模型结构测试、聚合函数测试及单客户端训练逻辑 2025-10-19 21:31:08 +08:00
TY1667
314f46d542 重构fed_run.py,移除冗余函数,传参BUG修复,更新模型权重保存逻辑;新增fed_run.sh脚本以支持分布式训练 2025-10-19 21:30:45 +08:00
TY1667
0343a0fd30 新增__init__.py文件,fed_util.py 结构优化 2025-10-19 21:29:58 +08:00
TY1667
3f4dd07572 优化util.py,简化代码结构,修复非极大值抑制和计算AP函数中的张量维度问题 2025-10-19 21:28:25 +08:00
TY1667
300ce2e93f 优化Dataset类,简化参数处理,修复图像加载错误处理,移除冗余代码 2025-10-19 21:28:04 +08:00
TY1667
40de29591b 优化FedYoloClient和FedYoloServer类 2025-10-19 21:27:19 +08:00
101ffa51eb 添加coco128配置文件 2025-10-03 20:25:09 +08:00
beaa290c19 更新README.md,添加运行说明和待办事项列表 2025-10-03 20:24:10 +08:00
86c7579b42 修复非极大值抑制函数中的张量维度问题,并更新compute_ap函数以使用numpy.trapezoid替代已弃用的numpy.trapz 2025-10-03 20:23:59 +08:00
33586e0c0c 优化test和agg方法,增强模型评估和聚合逻辑的稳定性 2025-10-03 20:23:50 +08:00
9a5e6b5b71 优化fed_run函数中的进度条显示和训练过程中的日志记录 2025-10-03 20:23:34 +08:00
964a8024c0 优化fed_run函数中的进度条显示 2025-10-02 22:37:50 +08:00
0b52cfc4f5 修改聚合逻辑,每次聚合不再创建新模型 2025-10-02 22:37:22 +08:00
c2e538898c 完善_parse_yolo_label_file函数的文档字符串,添加参数和返回值说明 2025-10-02 22:34:29 +08:00
TY1667
b19f11125d 更新.gitignore,添加对results/目录和所有.log文件的忽略 2025-10-02 16:38:54 +08:00
TY1667
1ae76d0aed Fedavg and YOLOv11 training 2025-10-02 16:31:55 +08:00
a60e002733 Initial commit 2025-10-02 08:19:50 +00:00