feat(yolo): 添加刀具识别AI模型训练与自动标注功能

Browse Source

- 新增 README.md 文档，包含完整的 YOLOv8 训练环境搭建指南
- 添加 docker-compose.yml 配置文件，支持 Mac M系列环境下的容器化部署
- 实现 train_and_export.py 核心脚本，集成模型训练、验证和ONNX格式导出功能
- 创建 data.yaml 数据集描述文件，定义刀具分类(普通刀具和管制刀具)
- 开发 auto_label.py 自动标注脚本，支持AI辅助的数据集标注处理
- 集成 YOLOv8 中型模型，优化标注准确率和识别能力
- 添加数据集验证和可视化功能，便于标注质量检查

...

This commit is contained in:

yuzl6

2026-03-10 17:15:05 +08:00

commit 1753cafaab

49 changed files with 331 additions and 0 deletions

Show all changes Ignore whitespace when comparing lines Ignore changes in amount of whitespace Ignore changes in whitespace at EOL

Download Patch File Download Diff File Expand all files Collapse all files

datasets/raw_normal/105d037e9c1f1cc8a7faa4bbd698ef5a~tplv-be4g95zd3a-image.jpeg

BIN

Binary file not shown.

Side by Side After Width:  |  Height:  |  Size: 358 KiB