若该文为原创文章，转载请注明原文出处。
转换成onnx模型(windows)
一、查看pt文件
准备好训练好的pt文件,可以用Netron打开看看大概长啥样:
二、模型转换
主要的目的是想把训练好的pt文件转成onnx模型，为后面RV1126的部署做铺垫。
我们是在py38的conda环境, 进行转换的，所以先切换到 yolo环境，并进入到yolov5目录下
之后输入转换命令
python export.py --rknpu rv1126 --weight runs/train/exp4/weights/best.pt --include onnx
需要注意的是pt文件的路径，命令的参数可以在README里自行了解。
不出意外，会在pt的目录下生成onnx模型
用Netron看看模型
三、总结
1、模型转换一定要在py3.8环境下运行。
如有侵权，请及时联系博主删除。
Original: https://blog.csdn.net/weixin_38807927/article/details/128430088
Author: 殷忆枫
Title: RV1126笔记十六：吸烟行为检测及部署＜四＞
原创文章受到原创版权保护。转载请注明出处：https://www.johngo689.com/795485/
转载文章受原作者版权保护。转载请注明原作者出处！