|
|
6a772b23ed
|
feat: 添加 YOLOv12 模型继续训练脚本
- 新增 continue_train.py 文件，实现从指定模型恢复训练状态并继续训练的功能
- 支持通过命令行参数指定预训练模型路径、数据配置文件、训练轮数等参数
-增加了对最佳权重的加载逻辑，确保从最优模型开始继续训练
- 强制使用 CPU 进行训练，避免 CUDA 设备错误
|
2025-06-04 23:24:53 +08:00 |
|
|
|
f13e50aa6b
|
feat add tflite model
|
2025-06-04 03:17:23 +08:00 |
|
|
|
3ba3097b6f
|
Changes fix model transfer bugs
|
2025-06-04 02:20:09 +08:00 |
|
|
|
7e38b3acfd
|
feat: 添加 YOLO 模型从 PyTorch 到 ONNX 的转换及验证功能
- 实现了将 YOLO模型从 PyTorch 格式转换为 ONNX 格式的功能
- 加载并验证了导出的 ONNX 模型
- 编写了使用验证集进行模型测试的代码，包括图像预处理和推理
- 添加了调试信息输出，便于分析模型性能和验证结果
|
2025-06-03 23:49:21 +08:00 |
|
|
|
5f7e66e97f
|
feat(datasets): 添加新的训练和验证标签文件
- 在 train 和 valid 文件夹下添加了多个新的标签文件
- 标签文件包含不同类别的对象标注信息- 新增的标签文件将有助于模型训练和评估
|
2025-06-03 23:13:32 +08:00 |
|
