yolov8存档

# main.py

from ultralytics import YOLO

model = YOLO(r'ultralytics\cfg\models\v8\yolov8-CBAM.yaml')

model.train(**{'cfg':'ultralytics\cfg\default.yaml', 'data':'data.yaml'}) 

# val.py

from ultralytics import YOLO
 
model = YOLO('runs/detect/train5(v2)/weights/best.pt')

model.val(data='data.yaml',batch=1,workers=0)

# test.py

from ultralytics import YOLO
 
model = YOLO('runs/detect/train5(v2)/weights/best.pt')

model.predict(source=r'datasets\wind turbine.v2\test\images',save=True)

# data.yaml

path: wind turbine.v2
train: train/images
val: valid/images
test: test/images

nc: 2
names: ['damage', 'dirt']