From 1d3880e1fd242b05bfc3d19c6eead313b8aa04e1 Mon Sep 17 00:00:00 2001
From: mshe <666666666@666666666.666666666>
Date: Sat, 27 Sep 2025 18:04:08 +0800
Subject: [PATCH] bak

---
 .DS_Store               | Bin 0 -> 6148 bytes
 14.训练模型-姿态估计.py |  64 +++++++++++++++++++---------------------
 2 files changed, 31 insertions(+), 33 deletions(-)
 create mode 100644 .DS_Store

diff --git a/.DS_Store b/.DS_Store
new file mode 100644
index 0000000000000000000000000000000000000000..d2402703793529d3a605f2ad3423ca0564258020
GIT binary patch
literal 6148
zcmeH~F$w}f3`G;&La^D=avBfd4F=H@>;)_Y8(CpR>p8kVnIJe?i^vZof0CK7>??LQ
zBBI;daU;@;$O3LEYYUAj@=hkX$tc(RFpcNSctXwUvp8#OcrVR<Y*R>p1W14cNPq->
zh=ASOu*I?}BMFcI2|Nkd_aVVeb7-mRpAG~c0iYeqZdm&)0Zmqb=Fn0V6`02KpwX&6
zhFIO(p()Pg&{DOwi^lMw@np3r2Bx)LG$Da$c3~g^5@-m_Yu?-azlVRC|C<)3BtQax
zMnJpWu-oFL;%<F=J*yw0YU=`r`f-Gpp8zCw6|dlK*iW{A=Fn0V6&Qa690LOhe3ifh
D(IFDs

literal 0
HcmV?d00001

diff --git a/14.训练模型-姿态估计.py b/14.训练模型-姿态估计.py
index 0cdf754..2d6f410 100644
--- a/14.训练模型-姿态估计.py
+++ b/14.训练模型-姿态估计.py
@@ -1,7 +1,7 @@
 from ultralytics import YOLO
 import torch
 # 加载预训练模型
-# model = YOLO('models/yolov8n-pose.pt')
+model = YOLO('models/yolov8n-pose.pt')
 # model = YOLO('models/yolo11x-pose.pt')
 # model = YOLO('models/yolov8n-pose.pt')
 # model = YOLO('runs/pose/train2/weights/last.pt')
@@ -22,38 +22,36 @@ import torch
 # save_period: 每隔多少轮保存一次模型
 # save_dir: 训练结果保存路径
 # weights: 预训练模型路径
-# model.info()
-# model.train(data='./dataset1/train.yaml', epochs=300, imgsz=640, batch=32, device='cpu')
+model.info()
+model.train(data='./dataset1/train.yaml', epochs=300, imgsz=640, batch=32, device=0)
 
 
 
-
-# 检查MPS可用性
-print(f"MPS available: {torch.backends.mps.is_available()}")
-print(f"MPS built: {torch.backends.mps.is_built()}")
-
-model = YOLO('models/yolov8n-pose.pt')
-
-# MPS优化配置
-model.train(
-    data='./dataset1/train.yaml',
-    epochs=300,
-    imgsz=320,  # M2 Pro上建议减小尺寸
-    batch=8,    # 根据内存调整
-    device='mps',  # 使用Apple Metal Performance Shaders
-    workers=2,  # MPS下建议2个worker
-    patience=50,
-    lr0=0.01,
-    lrf=0.01,
-    momentum=0.9,  # MPS上动量稍小
-    weight_decay=0.0005,
-    warmup_epochs=5.0,
-    box=7.5,
-    pose=1.0,   # 增加姿态损失权重
-    kobj=1.5,
-    save=True,
-    exist_ok=True,
-    verbose=True,
-    amp=False   # MPS上关闭自动混合精度
-)
-print("训练完成")
+#
+# # 检查MPS可用性
+# print(f"MPS available: {torch.backends.mps.is_available()}")
+# print(f"MPS built: {torch.backends.mps.is_built()}")
+#
+# model = YOLO('models/yolov8n-pose.pt')
+#
+# # MPS优化配置
+# model.train( data='./dataset1/train.yaml', epochs=300,
+#     imgsz=320,  # M2 Pro上建议减小尺寸
+#     batch=8,    # 根据内存调整
+#     device='mps',  # 使用Apple Metal Performance Shaders
+#     workers=2,  # MPS下建议2个worker
+#     patience=50,
+#     lr0=0.01,
+#     lrf=0.01,
+#     momentum=0.9,  # MPS上动量稍小
+#     weight_decay=0.0005,
+#     warmup_epochs=5.0,
+#     box=7.5,
+#     pose=1.0,   # 增加姿态损失权重
+#     kobj=1.5,
+#     save=True,
+#     exist_ok=True,
+#     verbose=True,
+#     amp=False   # MPS上关闭自动混合精度
+# )
+# print("训练完成")