# coding:utf-8
from ultralytics import YOLO

# 模型配置文件,此处以 s 为例，只需要写 yolo11s.yaml ，yolo11.yaml 会自动定位为 yolo11s 模型
# 如果写成yolo11m.yaml ，yolo11.yaml 会自动定位为 yolo11m 模型
model_yaml_path = r"ultralytics/cfg/models/11/MobileNetV4.yaml"
# 数据集配置文件
data_yaml_path = r'/home/hx/yolo11-jz/ailaidata.yaml'

# 预训练模型
pre_model_name = 'yolo11n.pt'

if __name__ == '__main__':
    # 加载预训练模型
    model = YOLO(model_yaml_path, task='detect').load(pre_model_name)
    # 训练模型
    results = model.train(
        data=data_yaml_path,
        epochs=100,
        batch=16,
        name='train_emoKDEF_MobileNetV4_s',
    )