点击上方蓝字关注我们

微信公众号:OpenCV开发者联盟

关注获取更多计算机视觉与深度学习知识

问题

    当你使用YOLOv8命令行训练模型的时候,如果当前执行的目录下没有相关的预训练模型文件,YOLOv8就会自动下载模型权重文件。这个是一个正常操作,但是你还会发现,当你在参数model中指定已有的,在其他目录下的预训练模型文件,YOLOv8还是会一样去下预训练模型文件,直接无视model参数指定,于是多数开发者都是确保在当前目录下有YOLOv8模型预训练文件,然后再从当前目录下执行下面的命令行训练:
yolo task=detect mode=train model=yolov8n.pt epochs=5 batch=1 data=D:\python\my_yolov8_train_demo\dm_dataset.yaml

解决

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


修改配置


常用的配置选项支持:


简单粗暴,直接修改配置文件拉倒了,修改好了以后,我真的可以了

C:\Users\Administrator\AppData\Roaming\Ultralytics\settings.yaml
直接改上面这个文件内容,改完保存一下即可!再重启命令行,就可以从任何目录下开启YOLOv8模型训练,它再也不会每次都去下载预训练模型文件了。也不会到处乱保存模型训练结果了。


这次是真的可以了。

推荐阅读

用注意力机制魔改YOLOv5涨点

一个教程搞定YOLOv8 对象检测+跟踪+分割


扫码查看OpenCV+OpenVIO+Pytorch系统化学习路线图