!cd yolov5 && python train.py –img 320 –batch 16 –epochs 1500 –data dataset.yml –weights yolov5s.pt –workers 2
This is for training the yolov5 model to recognize the labels “awake” and “drowsy” via image detection but it cannot detect any image files in the data/images folder. This is for vscode python notebook. The dataset.yml file is as follows:
path: ../data
train: images
val: images
#Classes
nc: 17
names: [‘dog’,
‘person’,
‘cat’,
‘tv’,
‘car’,
‘meatballs’,
‘marinara sauce’,
‘tomato soup’,
‘chicken noodle soup’,
‘french onion soup’,
‘chicken breast’,
‘ribs’,
‘pulled pork’,
‘hamburger’,
‘cavity’,
‘awake’,
‘drowsy’
]
debug: True
debug_verbose: True
I am expecting the model to be properly trained and in the folder …Drowsiness Detection using YOLO v5Projectyolov5runstrain there are exp folders with best.pt and last.pt in weights folder
048 Elham Bhuiyan is a new contributor to this site. Take care in asking for clarification, commenting, and answering.
Check out our Code of Conduct.