-
e999853c94
fix pylance type warning
main
TY1667
2025-11-04 13:18:43 +08:00
-
f658c6aca5
Fix Error: AttributeError: 'collections.OrderedDict' object has no attribute 'half'
root
2025-11-03 21:12:35 +08:00
-
fdb70869f9
更新配置文件,移除 YOLO 版本实现计划,调整 UAV 和 COCO 配置中的数据量和批次大小
root
2025-10-31 13:14:47 +08:00
-
291b82bec3
优化训练过程,添加 CSV 日志记录,改进模型权重保存机制
root
2025-10-31 13:14:37 +08:00
-
c7afef2dc2
更新配置文件路径,优化文档字符串,增强代码可读性
root
2025-10-31 13:14:29 +08:00
-
194ca8ee31
优化文档字符串,明确参数说明
root
2025-10-31 13:14:21 +08:00
-
c7a3d34d88
配置更改
root
2025-10-23 13:07:55 +08:00
-
b81f33ad28
更新运行指令,使用 bash 脚本替代 python 脚本
root
2025-10-23 13:07:44 +08:00
-
f9588b74a8
测试代码逻辑优化
root
2025-10-23 13:07:34 +08:00
-
2fbb741d3f
支持后台运行并记录日志
root
2025-10-23 13:07:04 +08:00
-
1822aca36b
优化验证集文件读取逻辑
root
2025-10-23 13:06:38 +08:00
-
ac4af34802
暂时禁用缓存加载功能,移除缓存文件以避免潜在问题
root
2025-10-23 13:06:24 +08:00
-
52382e460d
移除训练过程中的调试打印语句
root
2025-10-23 13:06:13 +08:00
-
76d7149512
更新.gitignore,新增忽略结果、日志和权重文件;更新README.md,添加参考链接
TY1667
2025-10-19 21:31:18 +08:00
-
f23a22632f
新增testcode.py文件,包含模型结构测试、聚合函数测试及单客户端训练逻辑
TY1667
2025-10-19 21:31:08 +08:00
-
314f46d542
重构fed_run.py,移除冗余函数,传参BUG修复,更新模型权重保存逻辑;新增fed_run.sh脚本以支持分布式训练
TY1667
2025-10-19 21:30:45 +08:00
-
0343a0fd30
新增__init__.py文件,fed_util.py 结构优化
TY1667
2025-10-19 21:29:58 +08:00
-
3f4dd07572
优化util.py,简化代码结构,修复非极大值抑制和计算AP函数中的张量维度问题
TY1667
2025-10-19 21:28:25 +08:00
-
300ce2e93f
优化Dataset类,简化参数处理,修复图像加载错误处理,移除冗余代码
TY1667
2025-10-19 21:28:04 +08:00
-
40de29591b
优化FedYoloClient和FedYoloServer类
TY1667
2025-10-19 21:27:19 +08:00
-
101ffa51eb
添加coco128配置文件
root
2025-10-03 20:25:09 +08:00
-
beaa290c19
更新README.md,添加运行说明和待办事项列表
root
2025-10-03 20:24:10 +08:00
-
86c7579b42
修复非极大值抑制函数中的张量维度问题,并更新compute_ap函数以使用numpy.trapezoid替代已弃用的numpy.trapz
root
2025-10-03 20:23:59 +08:00
-
33586e0c0c
优化test和agg方法,增强模型评估和聚合逻辑的稳定性
root
2025-10-03 20:23:50 +08:00
-
9a5e6b5b71
优化fed_run函数中的进度条显示和训练过程中的日志记录
root
2025-10-03 20:23:34 +08:00
-
964a8024c0
优化fed_run函数中的进度条显示
root
2025-10-02 22:37:50 +08:00
-
0b52cfc4f5
修改聚合逻辑,每次聚合不再创建新模型
root
2025-10-02 22:37:22 +08:00
-
c2e538898c
完善_parse_yolo_label_file函数的文档字符串,添加参数和返回值说明
root
2025-10-02 22:34:29 +08:00
-
b19f11125d
更新.gitignore,添加对results/目录和所有.log文件的忽略
TY1667
2025-10-02 16:38:54 +08:00
-
1ae76d0aed
Fedavg and YOLOv11 training
TY1667
2025-10-02 16:26:27 +08:00
-
a60e002733
Initial commit
Anchor-x
2025-10-02 08:19:50 +00:00