Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

能给个yolov7官方训练版本的转换流程吗,如果只是转好的onnx不能实际项目用 #11

Open
xs818818 opened this issue Jul 23, 2022 · 3 comments

Comments

@xs818818
Copy link

谢谢

@Spyderm4n
Copy link

官方的模型需要修改这位大佬的ORT代码才能实现推理。有两点需要注意:

  1. 大佬这里的预处理用的是 resize 而不是 padding ,改成 padding 以后就行了;
  2. 官方的模型转换的脚本可以直接带上 --grid --simplify --end2end,后处理可以省略非常多步骤,很方便,但是相应的也需要修改这位大佬的后处理部分代码,去掉NMS等部分 (因为官方v7带上--end2end 后出来的结果就直接是 bach_id, x1, y1, x2, y2, cls_id, prob 的结构)。

@BandyKenny
Copy link

2. 处理可以省略非常多步骤,很方便,但是相应的也需要修改这位大佬的后处理部分代码,去掉NMS等部分 (因为官方v7带上--end2end 后出来的结果就直接是

我利用官方的加 --grid --simplify --end2end转出来后的onnx是这样子的:
image
然后参考大佬说的,但是推理还是有问题,推理那一行可以正常执行,但是推理后得到的ort_output这个结果怎么取得到呀:
image
我这里debug就是显示[0]里面有地址。。。。

@Chenhaolin6
Copy link

python export.py --weights .\weights\best.pt --grid --topk-all 100 --iou-thres 0.65 --conf-thres 0.35 --img-size 640 640 --max-wh 640

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

4 participants