mshe
|
9504085c98
|
feat(pose): 添加姿态估计模型训练和检测脚本- 新增 YOLO 姿态估计模型训练脚本 (14.训练模型-姿态估计.py)
- 新增 YOLO 姿态估计模型检测脚本 (15.模型检测-姿态估计.py)- 添加 LabelMe 标注文件转 YOLO 格式的转换脚本 (labelme_to_yolo.py)
- 添加 LabelMe 姿态标注转 YOLO 姿态格式的转换脚本 (labelme_to_yolo_pose.py)- 新增训练数据集配置文件 (dataset1/train.yaml, dataset1/data.yaml)
- 添加训练样本标签文件 (dataset1/train/labels/5_png.rf.xxx.txt 等)
- 配置手部关键点检测参数，支持11个关键点- 设置图像尺寸为640，批次大小为32，训练轮数为30
- 支持CPU设备训练，预留GPU训练接口
- 实现关键点标签转换，包括归一化坐标处理
- 添加类别名称和关键点顺序定义
- 集成Roboflow数据集配置信息
- 支持模型推理视频和图片文件夹输入-保存检测结果到本地文件
- 提供模型加载和参数配置说明- 支持best.pt和last.pt模型权重加载
- 实现关键点可见性标注 (2 表示可见)- 添加关键点标签映射和排序逻辑
- 创建输出目录自动检查和创建机制- 输出转换完成提示信息
- 定义关键点标签顺序 (f_1_1, f_1_2, ..., f)
|
2025-09-27 02:54:06 +08:00 |