This commit is contained in:
mshe 2025-09-27 18:13:19 +08:00
parent 9ff6d23c77
commit 3103e78488
1 changed files with 2 additions and 2 deletions

View File

@ -3,7 +3,7 @@ from ultralytics import YOLO
# 加载训练好的模型 # 加载训练好的模型
# best.pt: 最佳模型,适用于生产 # best.pt: 最佳模型,适用于生产
# last.pt: 最后一轮训练的模型,适用于继续训练 # last.pt: 最后一轮训练的模型,适用于继续训练
# yolo = YOLO('runs/pose/train/weights/best.pt') yolo = YOLO('runs/pose/train/weights/best.pt')
yolo = YOLO('models/yolo11n-pose.pt') # yolo = YOLO('models/yolo11n-pose.pt')
# yolo('resources/input.mp4', show=True, save=True) # yolo('resources/input.mp4', show=True, save=True)
yolo.predict('resources/姿态估计-自定义', show=False, save=True) yolo.predict('resources/姿态估计-自定义', show=False, save=True)