I am trying to use my trained yolov5 model with the android app example here:
Android Object Detection Example
I figured i have to modify the values inside
PrePostProcessor.java
so i updated them according to my model
// model input image size
static int mInputWidth = 640;
static int mInputHeight = 640;
// model output is of size 25200*11
private static int mOutputRow = 25200; // as decided by the YOLOv5 model for input image of size 640*640
private static int mOutputColumn = 11; // left, top, right, bottom, score and 6 class probability
private static float mThreshold = 0.50f; // score above which a detection is generated
private static int mNmsLimit = 15;
........................................................
Result result = new Result(cls, outputs[i*11+4], rect);
This is how my input and output look like
My result view always places all predictions in the corner of the image (I cant upload multiple pictures)
Is there anyting else i should change?