yolov7 icon indicating copy to clipboard operation
yolov7 copied to clipboard

the exported onnx for yolo_pose seems not right

Open zhaoxuejun1234 opened this issue 1 year ago • 1 comments

The provided yolov7-w6-pose.pt is well used. I tried to export the onnx model with python export.py --weights 'yolov7-w6-pose.pt' --simplify I tried to infer the model with the exported model, the ouptput of the onnx is different from the "yolov7-w6-pose.pt" inference.The expected results are shown as image The given result of the onnx is shown as Snipaste_2022-08-01_18-27-12

zhaoxuejun1234 avatar Aug 01 '22 10:08 zhaoxuejun1234

It looks like the given export.py doesn‘t include the keypoints detection Class, directly exporting the ONNX from the pt file doesn;t allow the onnx node operation. I have to look into the model definition more clearly before the repo updated

zhaoxuejun1234 avatar Aug 02 '22 08:08 zhaoxuejun1234

Hi, you may want to try my exported model here

trancongman276 avatar Aug 17 '22 05:08 trancongman276

Hi, you may want to try my exported model here

how to export the model?

xinsuinizhuan avatar Aug 19 '22 00:08 xinsuinizhuan

@xinsuinizhuan 也可以看一下我这里的 https://github.com/BaofengZan/yolov7-pose-e2e-trt

BaofengZan avatar Aug 19 '22 01:08 BaofengZan

@xinsuinizhuan 也可以看一下我这里的 https://github.com/BaofengZan/yolov7-pose-e2e-trt

It a great work, the exported is perfect. How about the yolov7_mask, could you also exporte it?

xinsuinizhuan avatar Aug 22 '22 01:08 xinsuinizhuan

Hi, you may want to try my exported model here

how to export the model?

Instead of following Yolo repo's instruction, I export the model normally from PyTorch -> ONNX and it works like a charm.

trancongman276 avatar Aug 22 '22 03:08 trancongman276

Hi, you may want to try my exported model here hearts

how to export the model?

Instead of following Yolo repo's instruction, I export the model normally from PyTorch -> ONNX and it works like a charm.

Hello, I have reviewed your repository and it looks quite good, I tried it and it works perfectly, could you tell me how to obtain the weights in onnx that are in your repository from the yolov7-w6-pose.pt file? I have tried using the export.py and also following the instructions in the pytorch documentation but no success. Could you please share your conversion script or a link to a tutorial that I can follow. Thank you very much!

sperezs95 avatar Nov 01 '22 16:11 sperezs95

Hi, is anybody able to convert the yolov7-w6-pose.pt file to onnx and then to TensorRT? there is some confusion between using the branch 'pose' in this repo and to using other repos.

JJLimmm avatar Jan 18 '23 09:01 JJLimmm

Hi, you may want to try my exported model here

Hi, I am trying to use your solution but when I try to convert your onnx to engine on the Jetson platform I get this error: image

Could you help me?

MarcoPrassel avatar Mar 23 '23 16:03 MarcoPrassel