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模型