bak
This commit is contained in:
parent
9ff6d23c77
commit
3103e78488
|
|
@ -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)
|
||||||
Loading…
Reference in New Issue