原文:Yolov8 姿态估计 - 知乎 (zhihu.com)
YOLOv8论文还没有,官方默默又加了新模型:姿态估计。
现在你可以用YOLOv8做目标检测、实例分割、图像分类、目标跟踪、姿态估计,未完待续。。。。。。
一、Yolov8姿态估计
Yolov8的姿态估计模型是在COCO数据集训练的,目前支持的是人体的检测和姿态估计。
测试一下:
yolo pose predict model=yolov8n-pose.pt source='bus.jpg' show=True save=True
其中pose指定任务类型,predict代表我们是要做推断,模型这里我选择的是最轻量级的YOLOv8n-pose,”show=True save=True“代表显示并保存。运行的话,程序会自动下载需要的模型。
如果要用YOLOv8调用摄像头的话,命令如下:
yolo pose predict model=yolov8n-pose.pt source=0 show=True save=True
视频的话同理,只需将source=0 改为视频的路径即可。下面看一下效果:
00:30
二、模型训练:
按照coco128-pose.yaml的"样例"组织好数据并修改coco128-pose.yaml后,你只需要一句命令即可开始训练:
yolo pose train data=coco128-pose.yaml model=yolov8n-pose.pt epochs=100 imgsz=640
查看下coco128-pose.yaml文件