|
959a2b37c7
|
docs(runs): 添加最终模型训练结果分析报告
- 新增 final_model/analysis.md 文件
- 分析了损失函数、精度指标和学习率调整的趋势
- 总结了模型训练过程中的表现,指出了可能的过拟合问题
- 提出了改进建议,包括关注验证集表现和调整模型复杂度
|
2025-05-30 02:17:53 +08:00 |
|
|
72a61d34cd
|
build: 更新 ONNX 构建指南并添加训练配置文件
- 更新 README.md 中的 ONNX构建指南,增加新版本的安装步骤
- 在 runs/detect/final_model 和 runs/detect/train 目录下添加训练配置文件 args.yaml
- 配置文件包含模型训练的各种参数设置,为后续训练提供详细配置
|
2025-05-30 02:15:36 +08:00 |
|
|
28dad9e675
|
docs: 添加 ONNX 构建说明
添加了 ONNX 构建过程中可能遇到的问题及解决方法,具体包括:
- 克隆 ONNX 仓库
- 更新子模块
- 设置 CMAKE_ARGS 参数
- 使用 pip 安装
|
2025-05-29 23:08:56 +08:00 |
|
|
86ccd7c29e
|
refactor(train): 注释模型导出相关代码
- 注释掉了 opset_version 和 simplify 参数
- 保留了 format 和 dynamic 参数
|
2025-05-29 21:06:56 +08:00 |
|
|
21d190a658
|
train: 更新类别权重设置
- 从配置中移除了特定于绝缘子的类别权重设置
- 替换为更通用的类别权重配置,增加了一个类别权重- 保持了权重设置的注释形式,以便未来可能的启用
|
2025-05-29 20:23:21 +08:00 |
|
Asuka
|
66d0877cfd
|
refactor(data): 更新数据集路径并调整数据验证逻辑
- 修改 data.yaml 中的训练、验证和测试数据集路径
- 更新 train.py 中的数据验证逻辑,使用新的数据集路径
|
2025-05-29 20:16:02 +08:00 |
|
|
b4468c7808
|
feat(datasets): 添加新的训练数据标签
- 在 datasets/train/labels 目录下新增了多个标签文件
- 文件命名格式为:视频ID-帧序号.jpg.rf.哈希值.txt
-标签内容包括物体类别和位置信息,用于训练目标检测模型
|
2025-05-29 20:00:55 +08:00 |
|
|
20b660c943
|
feat(datasets): 添加新的训练数据标签
- 在 datasets/train/labels 目录下新增了多个标签文件
- 文件命名格式为:视频ID-帧序号.jpg.rf.哈希值.txt
-标签内容包括物体类别和位置信息,用于训练目标检测模型
|
2025-05-29 19:59:57 +08:00 |
|