Updates for YOLOv5 about onnxruntime-web-demo HOT 4 OPEN

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.

from onnxruntime-web-demo.

from onnxruntime-web-demo.

@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：

from onnxruntime-web-demo.

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

Thank you so much for your time.

from onnxruntime-web-demo.