yolov7-opencv-onnxrun-cpp-py icon indicating copy to clipboard operation
yolov7-opencv-onnxrun-cpp-py copied to clipboard

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

Open xs818818 opened this issue 2 years ago • 3 comments

谢谢

xs818818 avatar Jul 23 '22 03:07 xs818818

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

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

Spyderm4n avatar Aug 08 '22 07:08 Spyderm4n

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

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

BandyKenny avatar Aug 11 '22 09:08 BandyKenny

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

Chenhaolin6 avatar Aug 19 '22 03:08 Chenhaolin6