将pt文件转换成onnx文件

将pt文件转换成onnx文件_第1张图片

将pt文件转换成onnx文件

打开yolov5自带的export.py文件 更改weights信息
将pt文件转换成onnx文件_第2张图片
更改完成后在终端输入

python export.py --include torchscript onnx

如此即可实现pt到onnx的转换,路径在pt文件同级
将pt文件转换成onnx文件_第3张图片
如果要生成engine(用于tensorRT加速)格式则输入

python export.py --include engine --device 0

将pt文件转换成onnx文件_第4张图片

你可能感兴趣的:(yolov5,YOLO)