yolov7-opencv-onnxrun-cpp-py
yolov7-opencv-onnxrun-cpp-py copied to clipboard
能给个yolov7官方训练版本的转换流程吗,如果只是转好的onnx不能实际项目用
谢谢
官方的模型需要修改这位大佬的ORT代码才能实现推理。有两点需要注意:
- 大佬这里的预处理用的是 resize 而不是 padding ,改成 padding 以后就行了;
- 官方的模型转换的脚本可以直接带上 --grid --simplify --end2end,后处理可以省略非常多步骤,很方便,但是相应的也需要修改这位大佬的后处理部分代码,去掉NMS等部分 (因为官方v7带上--end2end 后出来的结果就直接是 bach_id, x1, y1, x2, y2, cls_id, prob 的结构)。
2. 处理可以省略非常多步骤,很方便,但是相应的也需要修改这位大佬的后处理部分代码,去掉NMS等部分 (因为官方v7带上--end2end 后出来的结果就直接是
我利用官方的加 --grid --simplify --end2end转出来后的onnx是这样子的:
然后参考大佬说的,但是推理还是有问题,推理那一行可以正常执行,但是推理后得到的ort_output这个结果怎么取得到呀:
我这里debug就是显示[0]里面有地址。。。。
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