Skip to content

Latest commit

 

History

History
executable file
·
30 lines (21 loc) · 1.55 KB

YOLOv7_Export_Guide.md

File metadata and controls

executable file
·
30 lines (21 loc) · 1.55 KB

YOLOv7模型导出

1. 准备工作

YOLOv7模型导出是在Pytorch模型的生产环境下进行的,需提前根据yolov7官方开源仓库的要求安装好Pytorch环境,准备好相应的代码和模型,并保证模型能够在Pytorch环境下正常推理运行。

注意:建议使用1.8.0+cpu的torch版本,避免因pytorch版本导致模型编译失败。

2. 主要步骤

2.1 导出torchscript、onnx模型

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模型所在目录下生成导出的torchscriptonnx模型,导出后可以修改模型名称以区分不同版本和输出类型,如yolov7_v0.1_3output.torchscript.pt ,yolov7_v0.1_3output.onnx表示带有3个输出的模型。

注意: 导出的torchscript模型建议以.pt为后缀,以免在后续模型编译量化中发生错误。

3. 常见问题

TODO