Thank you for your reply. I changed the value as calculated by 25200*(num_of_class+5) but I’m still facing the same issue.
Do I need to change this value also or any other ?
“private static int mOutputColumn = 85; // left, top, right, bottom, score and 80 class probability”
I changed only " private static int mOutputRow = 25200; // as decided by the YOLOv5 model for input image of size 640*640" this value.
Update:
I even changed mOutputColumn = 85 to my custom class number and class number + 5 also but I am still facing the same issue.