Training my YOLOv2 object detector failed
5 views (last 30 days)
I have created my network using the pre-trained network resnet18, and with the following code:
numClases = 9;
tamanio = [224 224 3];
anchorBoxes = estimateAnchorBoxes(TrainingEscalado,numClases)
red = resnet18();
arquitectura = yolov2Layers(tamanio,numClases,anchorBoxes,red,"res5b_relu","ReorgLayerSource","res3a_relu")
"TrainingEscalado" is the datastore for training already scaled to network's inputs size. When I import the network into the Deep Network Designer app, I don't have errors, and the same happens when I import the datastore for training.
But as I click train, an error message is displayed with the following content:
And I don't know what to do to fix it, the network already has 9 classes as output so I don't know what fails.