object-detection - YOLO物体检测模型微调
在自定义数据集上微调YOLO物体检测模型并导出为ONNX
标签
更新于: 2026-03-26该技能可以做什么
- 检查数据集格式
- 转换数据集格式
- 准备数据集划分
- 安装依赖包
- 训练YOLO模型
- 评估模型指标
- 导出ONNX模型
- 收集训练产物
- 生成训练总结
输入
- 自定义数据集目录
- 数据集配置文件
输出
- 训练好的模型权重
- 评估报告
- ONNX模型文件
- TensorBoard日志
- 可视化图表
要求
- Bash
- Python >= 3.8
- ultralytics
- onnx
- opencv-python-headless