When using the yolov5 tutorial, I have console logged the predection.shape in outputsToNMSPredictions, the shape of the example is [25200, 85].
When I use my custom trained model, the console logged prediction shape is [8400].
I believe this difference in the reason my model isn't working. It loads fine and the app doesn't crash, but it just comes back with an empty resultBoxes
Tutorial Select
Prepare Custom Model
Feedback
When using the yolov5 tutorial, I have console logged the
predection.shape
inoutputsToNMSPredictions
, the shape of the example is[25200, 85]
.When I use my custom trained model, the console logged prediction shape is
[8400]
.I believe this difference in the reason my model isn't working. It loads fine and the app doesn't crash, but it just comes back with an empty
resultBoxes
Does anybody have any advice? Thank you!