作者:殷忆枫

RV1126笔记十七:吸烟行为检测及部署<四>

若该文为原创文章,转载请注明原文出处。

转换成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环境下运行。

如有侵权,请及时联系博主删除。