yolov7-face icon indicating copy to clipboard operation
yolov7-face copied to clipboard

pt convert onnx ?? onnx convert ncnn

Open sssssshf opened this issue 2 years ago • 2 comments

1 请问pt转onnx的过程,我的命令是这样: python export.py --weights weights/yolov7-face.pt --img 640 --batch 1 --grid --simplify 转出来的模型 存在“slice”结构。 2 由于1中转出来的模型存在这个"slice" 结构,导致转ncnn一直失败。 请问您转onnx的代码 是怎么样的

sssssshf avatar Mar 30 '23 07:03 sssssshf

参考cpp目录下的代码

derronqi avatar Mar 31 '23 09:03 derronqi

我执行的是 1.python cpp/export.py --weights best.pt --img 640 --batch 1

2.python -m onnxsim best.onnx best_sim.onnx

然后用ncnn转best_sim.onnx 就可以了。但是我遇到ex.extract 会崩溃的问题

xll98 avatar Oct 12 '23 02:10 xll98