2.yolov8模型
下载链接,我下了yolov8s.py,放在该路径E:\nfshare\yolov8\ultralytics\weights
ps:model文件类型可以是yaml,也可以是pt
3.修改yolov8.yaml文件
E:\nfshare\yolov8\ultralytics\cfg\models\v8\yolov8.yaml,就改类别数
nc: 9 # number of classes
4.新建data文件
E:\nfshare\yolov8\ultralytics\cfg\datasets\hr.yaml,这些和yolov5一样
5.修改default.yaml文件
E:\nfshare\yolov8\ultralytics\cfg\default.yaml ,就动了以下几个参数
model: weights/yolov8s.pt # (str, optional) path to model file, i.e. yolov8n.pt, yolov8n.yaml
data: cfg/datasets/hr.yaml # (str, optional) path to data file, i.e. coco128.yaml
epochs: 1 # (int) number of epochs to train for
amp: False
batch: 8
运行代码 yolo cfg=cfg/default.yaml
6.训练yolov8模型(train)
1.新建.py文件训练模型
在路径下新建python脚本文件\yolov8\ultralytics\demo.py,就像运行yolov5的模型一样,运行该脚本文件。
下面这些参数要怎么设置我还没懂。
#import sys
#sys.path.append("/home/yyt/nfshare/yolov8/")
from ultralytics import YOLO
# Create a new YOLO model from scratch
#model = YOLO('/home/yyt/nfshare/yolov8/ultralytics/cfg/models/v8/yolov8.yaml')
# Load a pretrained YOLO model (recommended for training)
model = YOLO('/home/yyt/nfshare/yolov8/ultralytics/weights/yolov8s.pt')
# Train the model using the 'coco128.yaml' dataset for 3 epochs
results = model.train(data='/home/yyt/nfshare/yolov8/ultralytics/cfg/da