yolov5训练出来的pt转onnx
时间: 2023-07-29 21:13:02 浏览: 365
yolov5训练以及模型onnx转换
根据引用\[1\]中提供的信息,您可以按照以下步骤将yolov5训练出来的.pt模型转换为.onnx格式:
1. 在Ubuntu服务器上训练完模型并得到best.pt文件。
2. 在Windows系统下进行模型转换为.onnx格式的操作。
3. 首先备份export.py文件,并进行修改。
4. 打开chenexport.py文件,找到并修改模型位置的参数。将参数`--weights`的默认值修改为您的best.pt文件的路径,例如`--weights='../chenresult/best.pt'`。
5. 运行修改后的export.py文件,执行转换操作。
6. 在运行过程中,将会生成一个.onnx文件,即转换后的模型。
请注意,以上步骤仅适用于yolov5模型的转换,如果您使用的是其他模型,请参考相应的转换方法。
#### 引用[.reference_title]
- *1* [yolov5训练的结果转为onnx](https://blog.csdn.net/chencaw/article/details/118057505)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v91^control_2,239^v3^insert_chatgpt"}} ] [.reference_item]
[ .reference_list ]
阅读全文