YOLOv7模型导出是在Pytorch模型的生产环境下进行的,需提前根据yolov7官方开源仓库的要求安装好Pytorch环境,准备好相应的代码和模型,并保证模型能够在Pytorch环境下正常推理运行。
注意:建议使用
1.8.0+cpu
的torch版本,避免因pytorch版本导致模型编译失败。
YOLOv7不同版本的代码导出的YOLOv7模型的输出会有所不同,根据不同的组合可能会有1个、3个输出的情况,主要取决于导出模型的命令参数。YOLOv7官方仓库提供了模型导出脚本export.py
,可以直接使用它导出torchscript、onnx模型:
1输出的模型导出命令:
# 下述脚本可能会根据不用版本的YOLOv7有所调整,请以官方仓库说明为准
python3 export.py --weights yolov7.pt --grid
3输出的模型导出命令:
# 下述脚本可能会根据不用版本的YOLOv7有所调整,请以官方仓库说明为准
python3 export.py --weights yolov7.pt
上述脚本会在原始pt模型所在目录下生成导出的torchscript
、onnx
模型,导出后可以修改模型名称以区分不同版本和输出类型,如yolov7_v0.1_3output.torchscript.pt
,yolov7_v0.1_3output.onnx
表示带有3个输出的模型。
注意: 导出的torchscript模型建议以.pt
为后缀,以免在后续模型编译量化中发生错误。
TODO