cv516Buaa / tph-yolov5

GNU General Public License v3.0
716 stars 170 forks source link

onnx #51

Open largestcabbage opened 2 years ago

largestcabbage commented 2 years ago

How to convert TPH-yolo to onnx, why use your export.py to generate onnx, but cannot use

zhoufanking commented 1 year ago

I have the same problem, any advise?

sxstone commented 1 year ago

I have the same problem, any advise? +1

zhoufanking commented 1 year ago

I have the same problem, any advise? +1

it's due to the torch and torch vision version. You can try this torch==1.10.1 & torchvision ==0.11.2,also my onnx==1.12.0 onnxruntime==1.13.1.

sxstone commented 1 year ago

I have the same problem, any advise? +1

it's due to the torch and torch vision version. You can try this torch==1.10.1 & torchvision ==0.11.2,also my onnx==1.12.0 onnxruntime==1.13.1.

grateful!!!

leequan commented 1 year ago

I have the same problem, any advise? +1

it's due to the torch and torch vision version. You can try this torch==1.10.1 & torchvision ==0.11.2,also my onnx==1.12.0 onnxruntime==1.13.1.

grateful!!!

Can your onnx model be used in opencv of C++?

I have the same problem, any advise? +1

it's due to the torch and torch vision version. You can try this torch==1.10.1 & torchvision ==0.11.2,also my onnx==1.12.0 onnxruntime==1.13.1.

grateful!!!

Can your onnx model be used in opencv of C++?