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

import warnings
warnings.filterwarnings('ignore')
from ultralytics import YOLO

if __name__ == '__main__':
    model = YOLO(model=r'ultralytics/cfg/models/v13/yolov13n.yaml') # 模型配置文件 下面那个叫().pt：模型权重文件
    #model.load(r'') # 加载预训练权重,改进或者做对比实验时候不建议打开，因为用预训练模型整体精度没有很明显的提升
    # 如果需要加载权重，请指定有效路径，例如:
    #model.load(r'D:\yolov13\weights\yolov13.pt')
    model.train(data=r'ultralytics/cfg/datasets/yolov13.yaml',
                imgsz=640,
                epochs=500,
                batch=8,
                workers=0,
                device='',
                optimizer='SGD',
                close_mosaic=10,
                resume=False,
                project='runs/train',
                name='exp',
                single_cls=False,
                cache=False,
                )