（十二）rk3568 NPU 中部署自己训练的模型，（2）模型转换

对于rknn 模型部署，本人使用*.pt -> *.onnx -> *.rknn的方式。

一、首先是pt文件到onnx文件的转换。

onnx文件导出时，需要修改models/yolo.py文件中的后处理部分。

注意：在训练时不要修改yolo.py的这段代码，训练完成后使用export.py进行模型导出转换时一定要进行修改，不然会导致后面的rknn模型转换失败！  

 修改后将best.pt移动至export.py同一文件夹下，在命令行调用以下命令：

python export.py --weights best.pt --img 640 --batch 1 --include onnx

我们便可以得到成功转换的模型best.onnx。

二、模型网络结构查看

使用Netron工具可查看训练出的yolov5模型结构