在模型部署的过程中我们经常会进行下面这张图片(图片来源自openmmlab)所展示的流程:
所以接下来我们解决如何使用yolov5自带的export.py来进行onnx的转换。看到下面这串摘自export.py的代码:
'--include',
nargs='+',
default=['onnx'],
help='torchscript, onnx, openvino, engine, coreml, saved_model, pb, tflite, edgetpu, tfjs, paddle')
opt = parser.parse_args()
print_args(vars(opt))
return opt
可以看到default=['onnx'],而默认的是torchscript。有些人可能会通过终端运行
python export.py --include onnx
来获得onnx格式文件,我们完全可以使用export.py进行转换onnx甚至是其他格式文件。我的分享到此结束,谢谢!