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

描述

问题

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

yolo task=detect mode=train model=yolov8n.pt epochs=5 batch=1 data=D:pythonmy_yolov8_train_demodm_dataset.yaml

 

解决

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

检查配置

yolo settings

 

模型训练

修改配置

模型训练

常用的配置选项支持:

模型训练

模型训练

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

 

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

 

模型训练

这次是真的可以了。

 审核编辑:汤梓红

 

打开APP阅读更多精彩内容
声明:本文内容及配图由入驻作者撰写或者入驻合作网站授权转载。文章观点仅代表作者本人,不代表电子发烧友网立场。文章及其配图仅供工程师学习之用,如有内容侵权或者其他违规问题,请联系本站处理。 举报投诉

全部0条评论

快来发表一下你的评论吧 !

×
20
完善资料,
赚取积分