YOLO v3 - Error message while setup of training
2 次查看(过去 30 天)
显示 更早的评论
Hello,
i am trying to set up the training for a YOLO v3 - object detector. Everything is pretty much copied from the example website
Now, at the line:
[gradients, state, lossInfo] = dlfeval(@modelGradients, yolov3Detector, XTrain, YTrain,
penaltyThreshold);
I get the following error message, and i do not know how to fix this:
Check for incorrect argument data type or missing argument in call to function 'forward'.
Error in modelGradients (line 8)
dlYPred = forward(dlnetDiscriminator, dlX, dlT);
Does anything have an idea of how to fix this?
Thanks already in advance!
0 个评论
回答(1 个)
Maneet Kaur Bagga
2023-9-13
Hi Osman,
As per my understanding of the question, "dlfeval" is used as a handler for the modelGradients function which takes the "yolov3ObjectDetector" object, a mini-batch of input data XTrain with corresponding ground truth boxes YTrain, the specified penalty threshold as input arguments and returns the gradients of the loss with respect to the learnable parameters in "yolov3ObjectDetector", the corresponding mini-batch loss information, and the state of the current batch.
The error is due to the target data "dlT" paased as an input argument in the "forward" function, the forward function expects the input parameters to be the detector object and the training dataset and returns the output features of the network during the training phase. Removing the train data parameter “dlT” will resolve the error occured.
features = forward(detector,dlX);
Please refer to the following MATLAB Documentation for better understanding of the forward function:
Hope this helps!
Thank You
Maneet Bagga
0 个评论
另请参阅
Community Treasure Hunt
Find the treasures in MATLAB Central and discover how the community can help you!
Start Hunting!