search

pytorch / android-demo-app

PyTorch android examples of usage in applications
1.45k stars 593 forks source link

Yolov8 Object Detect Bounding box messed #321

Open Dieguinho5 opened 1 year ago

Dieguinho5 commented 1 year ago

Hello friends, I'm having a problem with my custom yolov8 template. The bounding boxes are messed up and it's giving a pretty low prediction. I'm using the starter project https://github.com/pytorch/android-demo-app/tree/master/ObjectDetection

Can anybody help me?

model yolov8.torchscript: https://drive.google.com/file/d/1-0IEYXOTyLkLlfrU6LeHy2SqJ7hh1g6G/view?usp=sharing image

janimo commented 1 year ago

In general yolo v5 and v8 have differently shaped outputs so dropping in a v8 model is not supposed to work out of the box.

encouver commented 11 months ago

In general yolo v5 and v8 have differently shaped outputs so dropping in a v8 model is not supposed to work out of the box.

Do you have some reference to use with yolov8?

NeighborhoodCoding commented 11 months ago

Can you please share the full code on how to change the yolov5 code to yolov8 code?