pose_train.py

# -*- coding: utf-8 -*-

"""
@Time    : 2025/2/02 16:31
@File    : train.py
@Author  : zj
@Description:

Usage - training using YOLOv8-pose / YOLO11-pose:
    $ python3 pose_train.py --model yolov8n-pose.pt --data ./yolo11face/cfg/datasets/widerface-landmarks.yaml --epochs 300 --imgsz 800 --batch 8 --device 0
                                    yolov8s-pose.pt
                                    yolo11n-pose.pt
                                    yolo11s-pose.pt

"""

import yolo11face_utils
from yolo11face_utils import parse_args

from ultralytics.models.yolo.pose import PoseTrainer


def main():
    overrides = parse_args()
    assert overrides['model'] is not None, 'model must be specified'
    assert overrides['data'] is not None, 'data must be specified'
    overrides['mode'] = 'train'

    # 初始化训练器并开始训练
    trainer = PoseTrainer(overrides=overrides)
    trainer.train()


if __name__ == "__main__":
    main()