YOLOv8实现任意目录下命令行训练

问题

当你使用YOLOv8命令行训练模型的时候,如果当前执行的目录下没有相关的预训练模型文件,YOLOv8就会自动下载模型权重文件。这个是一个正常操作,但是你还会发现,当你在参数model中指定已有的,在其他目录下的预训练模型文件,YOLOv8还是会一样去下预训练模型文件,直接无视model参数指定,于是多数开发者都是确保在当前目录下有YOLOv8模型预训练文件,然后再从当前目录下执行下面的命令行训练:

解决

无法从任意一个目录下发起YOLOv8模型训练是个问题!要解决这个问题其实很简单,YOLOv8是有个全局设置参数的,它支持从设置的目录下寻找预训练权重文件,同时支持把训练结果保存到指定目录下。这样设置好以后,我们就可以从任意目录下发起YOLOv8模型训练了,YOLOv8框架默认支持通过命令行修改这些设置参数的。

yolo settings

python 代码中修改

from ultralytics.utils import SETTINGS
print(SETTINGS["runs_dir"])
print(SETTINGS)
update_params = {'runs_dir': 'runs'}
SETTINGS.update(update_params)
print(SETTINGS)

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值