YOLOv8 训练预测更改save输出位置全网首发

TM 2.0

已于 2024-06-17 16:21:44 修改

阅读量1.1w

点赞数 29

文章标签： YOLO

于 2024-02-05 14:30:46 首次发布

本文链接：https://blog.csdn.net/qq_42693842/article/details/136041335

版权

本文详细解释了如何在Ultralytics的YOLO项目中修改default.yaml文件中的save_dir字段，以及如何自定义训练保存路径。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

方法 1

先给代码

from ultralytics.utils import DEFAULT_CFG
from datetime import datetime
current_time = datetime.now()
time_str = current_time.strftime("%Y-%m-%d_%H-%M-%S") # 个人习惯, 用训练时间命名保存路径, 或者你自己自定义
DEFAULT_CFG.save_dir= f"../models/{
     time_str}"
model = YOLO('yolov8n.pt')
data = 'data.yaml'
result = model.train(data = data, epochs = 100, batch = 32, save = True, device = 0