yolov11 python训练代码train.py

最新推荐文章于 2025-05-24 14:14:11 发布

我真不想CV

最新推荐文章于 2025-05-24 14:14:11 发布

阅读量228

点赞数 3

CC 4.0 BY-SA版权

文章标签： YOLO python 机器学习

本文链接：https://blog.youkuaiyun.com/xlxun/article/details/147642025

yolov11的训练代码，不多说直接付代码（train.py）

from ultralytics import YOLO

if __name__ == '__main__':
    model = YOLO(r'/root/ultralytics-main/ultralytics/cfg/models/11/yolo11.yaml')  # 此处以 m 为例，只需写yolov11m即可定位到m模型-SCI
    model.train(data=r'data.yaml',
                imgsz=320,
                epochs=300,
                batch=64,
                workers=4,
                device='0',
                patience=20,
                single_cls= False,
                )