search

WongKinYiu / yolov7

Implementation of paper - YOLOv7: Trainable bag-of-freebies sets new state-of-the-art for real-time object detectors
GNU General Public License v3.0
13.36k stars 4.22k forks source link

the exported onnx for yolo_pose seems not right #386

Open zhaoxuejun1234 opened 2 years ago

zhaoxuejun1234 commented 2 years ago

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 commented 2 years ago

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

trancongman276 commented 2 years ago

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

xinsuinizhuan commented 2 years ago

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

how to export the model?

BaofengZan commented 2 years ago

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

xinsuinizhuan commented 2 years ago

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

trancongman276 commented 2 years ago

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.

sperezs95 commented 2 years ago

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!

JJLimmm commented 1 year ago

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.

MarcoPrassel commented 1 year ago

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?