From ff389389dc0287f5c3195fe3c97d8f531aa7ad7b Mon Sep 17 00:00:00 2001 From: mshe <666666666@666666666.666666666> Date: Sat, 27 Sep 2025 18:16:55 +0800 Subject: [PATCH] bak --- 14.训练模型-姿态估计-windows.py | 44 ++++++++++++++++----------------- 1 file changed, 22 insertions(+), 22 deletions(-) diff --git a/14.训练模型-姿态估计-windows.py b/14.训练模型-姿态估计-windows.py index 1a10fc2..7c2e700 100644 --- a/14.训练模型-姿态估计-windows.py +++ b/14.训练模型-姿态估计-windows.py @@ -18,28 +18,28 @@ model.train( device=0, # 使用GPU 0 workers=8, # 充分利用CPU核心 patience=50, - lr0=0.01, - lrf=0.01, - momentum=0.937, - weight_decay=0.0005, - warmup_epochs=3.0, - - # 损失权重调整(重点解决姿态问题) - box=7.5, - pose=1.5, # 提高姿态损失权重 - kobj=2.0, # 提高关键点目标权重 - cls=1.0, - - # 性能优化 - amp=True, # 自动混合精度训练 - cos_lr=True, # 余弦学习率调度 - close_mosaic=10, # 最后10epoch关闭马赛克增强 - - # 数据增强 - hsv_h=0.015, - hsv_s=0.7, - hsv_v=0.4, - fliplr=0.5, # 水平翻转对姿态很重要 + # lr0=0.01, + # lrf=0.01, + # momentum=0.937, + # weight_decay=0.0005, + # warmup_epochs=3.0, + # + # # 损失权重调整(重点解决姿态问题) + # box=7.5, + # pose=1.5, # 提高姿态损失权重 + # kobj=2.0, # 提高关键点目标权重 + # cls=1.0, + # + # # 性能优化 + # amp=True, # 自动混合精度训练 + # cos_lr=True, # 余弦学习率调度 + # close_mosaic=10, # 最后10epoch关闭马赛克增强 + # + # # 数据增强 + # hsv_h=0.015, + # hsv_s=0.7, + # hsv_v=0.4, + # fliplr=0.5, # 水平翻转对姿态很重要 save=True, exist_ok=True,