craft-zhang/tensorflow-lite-cpp-examples

When i use yolov5 scripts to predict yolov8 model i am getting abnormal scores, predictions

Opened this issue · 4 comments

I trained a custom model and i am using your yolov5 scripts . The output tensor when i use yolov8 model is not matching with python version should i need to update your yolov5 script for yolov8 ? Also the scores, labels, boxes are going to 10000, 2000 ranges

Of course, the project is pretty ugly right now. The support for different versions of yolo is also messy. But thank you very much for your contribution and support.

thanks for your reply could you give me any links which would work for float32 model for yolov8

currently the scores, labels, and boxes are hard coded in the YOLO::run() method and for the input data format you can ref

} else if (_in_type == kTfLiteUInt8) {

okay thanks