Can Someone share their YOLO V3 or V4 with multiple classes.

Can anyone share their YOLO V3 or V4 with mutiple classes, I cannot get mine to train for the life of me, or it doesnt read the classes correctly. I think an example would help. The code and the workspace would let me decipher what I need to do.

Answer 1

[Nihal Reddy]

I understand that you want to train YOLO v3 or YOLO v4 on a dataset with multiple classes. In case of multiple classes data can be organized as three columns where the first column contains the image file names with paths, the second column contains the bounding boxes and the third column must be a cell vector that contains the label names corresponding to each bounding box.

For more information on how to arrange the bounding boxes and labels, refer the following link-

Datastore for bounding box label data - MATLAB (mathworks.com)

For more information regarding training YOLO v3, refer the following link-

Object Detection Using YOLO v3 Deep Learning - MATLAB & Simulink (mathworks.com)

For more information regarding training YOLO v4, refer the following link-

Object Detection Using YOLO v4 Deep Learning - MATLAB & Simulink (mathworks.com)