外贸快车智能建站,海外社交媒体营销,代运营网站建设,网站生成系统文章目录 1下载yolov8#xff08;[网址](https://github.com/ultralytics/ultralytics)#xff09;2用pycharm打开文件3训练自己的YOLOV8数据集4run下运行完了之后没有best.pt文件5导出为onnx文件 本章内容主要解决如何训练自己的YOLOV8模型。 1下载yolov8#xff08;网址[网址](https://github.com/ultralytics/ultralytics)2用pycharm打开文件3训练自己的YOLOV8数据集4run下运行完了之后没有best.pt文件5导出为onnx文件 本章内容主要解决如何训练自己的YOLOV8模型。 1下载yolov8网址 2用pycharm打开文件
关于环境配置参考-初始YOLOV8
在ultralytics中加入创建文件夹里面存档你的images、lables和yaml文件 images labels yaml文件这个是从coco128中改的coco128.yaml路径为ultralytics/cfg/datasets/coco8.yaml
3训练自己的YOLOV8数据集
打开终端 输入如下
yolo detect train datadatasets/code_inspection/code_inspection.yaml modelyolov8n.yaml pretrainedultralytics/yolov8n.pt epochs200 batch8 lr00.01 resumeTrue运行之后会出现一个询问wanna会询问你要不要使用wandb记录训练结果这个时候需要你点击进入链接创建一个wandb的账号账号会自动生成一个号码就可以用wandb了。
4run下运行完了之后没有best.pt文件
yolov8的训练的文件会放在网络端当然也是有办法放在本地的。我是没有的。 点击进入最上面的链接 按照如下顺序点击即可下载模型。反正我觉得不是很方便。 啊这过了个小时run文件下权重文件直接就出现了。可能是我重启之后的效果。
5导出为onnx文件
这个倒是很方便的。创建一个文件赋值下面代码就可以了。
from ultralytics import YOLO# Load a model
model YOLO(best.pt) # load a custom trained model# Export the model
model.export(formatonnx)
