YOLOv8基础知识 | 一文带你了解yolov8如何训练

最新推荐文章于 2025-01-17 23:05:36 发布

原创

最新推荐文章于 2025-01-17 23:05:36 发布 · 329 阅读

1 ·

CC 4.0 BY-SA版权

文章标签：

#YOLO #深度学习 #机器学习 #人工智能 #目标检测

模型训练

可以通过在根目录下创建一个train.py文件训练代码，在其中输入代码：

只需更换数据集路径以及yaml文件路径即可！

import sys
import argparse
import os


from ultralytics import YOLO


def main(opt):
    yaml = opt.cfg
    weights = opt.weights
    model = YOLO(yaml) # 直接加载yaml文件训练
    # model = YOLO(weights)  # 直接加载权重文件进行训练
    # model = YOLO(yaml).load(weights) # 加载yaml配置文件的同时，加载权重进行训练
    model.info()

    results = model.train(data=r'coco128.yaml', 
                        epochs=200, 
                        imgsz=640, 
                        workers=2, 
                        batch=2,
                        )

def parse_opt(known=False):
    parser = argparse.ArgumentParser()
    parser.add_argument('--cfg', type=str, default= r'ultralytics\cfg\models\v11\yolov11.yaml', help='initial weights path')
    parser.add_argument('--weights', type=str, default='yolov8n.pt', help='')

    opt = parser.parse_known_args