如何使用YOLO8训练自己的模型
以安全帽为例
数据网址
https://www.kaggle.com/datasets/snehilsanyal/construction-site-safety-image-dataset-roboflow
#safehat.py
from ultralytics import YOLO
model = YOLO('yolov8n.pt')
if __name__ == '__main__':
model.train(data='safehat.yaml',epochs=8)
model.val()
safehat.yaml
train: C:\GQ\opject\site_safty\01safehat\code\css_data\train\images
val: C:\GQ\opject\site_safty\01safehat\code\css_data\valid\images
test: C:\GQ\opject\site_safty\01safehat\code\css_data\test\images
nc: 10
names: [Hardhat, Mask, NO_Hardhat, NO-Mask, NO_Safety Vest, Person, Safety Cone,Safety Vest, machinery, vehicle]