Updates for YOLOv5

[jkassis]

Hi. Hope someone is still looking at this.

I updated this demo to use onnxruntime-web in my public fork... https://github.com/jkassis/onnxjs-demo.

I then extracted the yolo pieces to get this working with an onnx model of yolov5 exported from the ultralytics project: https://github.com/ultralytics/yolov5.

I'm having some problems with the post-processing phase as detailed in a ticket against the ultralytics repo...

Please see... ultralytics/yolov5#5595

Hoping that someone here might be able to take a look.

[Mr-LiuDC]

Hi, @jkassis

Have you solved your problems? I also met your issue ultralytics/yolov5#5595.

I trained my own custom dataset with yolov7 transfer learning and exported the onnx file, and also three tensors come out.

Any suggestions?

Thank you.

[jkassis]

Hi Leo, No. I never got the onnxruntime-web-demo working unfortunately. In fact, I never managed to get anything working end to end with onnx here.

…

On Mon, 30 Jan 2023 at 19:03, Leo ***@***.***> wrote: Hi, @jkassis <https://github.com/jkassis> Have you solved your problems? I also met your issue ultralytics/yolov5#5595 <ultralytics/yolov5#5595>. I trained my own custom dataset with yolov7 transfer learning and exported the onnx file, and also three tensors come out. Any suggestions? Thank you. — Reply to this email directly, view it on GitHub <#7 (comment)>, or unsubscribe <https://github.com/notifications/unsubscribe-auth/AA3WITF2GV276JI744M5CTTWVB6IPANCNFSM5H3JEPQA> . You are receiving this because you were mentioned.Message ID: ***@***.***>

[Mr-LiuDC]

@jkassis Thanks anyway.

BTW, If you are still interested in this. Here I have some experience maybe useful.

I use yolov7, here are to repositories:

Official: https://github.com/WongKinYiu/yolov7
Personal: https://github.com/bubbliiiing/yolov7-pytorch

onnxruntime-web detection example: https://github.com/Hyuto/yolov7-onnxruntime-web
and for you yolov5: https://github.com/Hyuto/yolov5-onnxruntime-web

Our onnx model firstly trained by bubbliiiing/yolov7-pytorch and come out three tensors, We need to concat the three outs into one, but I do't know how to do this. So, I change to official repo WongKinYiu/yolov7 and work well.

bubbliiiing/yolov7-pytorch trained custom dataset and export onnx：

WongKinYiu/yolov7 trained custom dataset and export onnx, works well：

[MarGraz]

Hi guys, @jkassis @Mr-LiuDC do you have any suggestion about this issue I opened?

Thank you so much for your time.