将onnx模型转化为RV1126平台的rknn模型

本文主要是介绍将onnx模型转化为RV1126平台的rknn模型，希望对大家解决编程问题提供一定的参考价值，需要的开发者们随着小编来一起学习吧！

在将onnx装成rknn模型之前，首先需要准备一些校准图片，比如你的模型时用来见得行人的，那么你可以准备100张行人图片放到文件夹中，然后用下面脚本将图片的路径生成到dataset.txt文件中

import osimages = "./test"
with open("dataset.txt",'w') as f:for image_name in os.listdir(images):f.write(os.path.join(images,image_name) + "\n")

生成后的txt内容如下图所示

 然后用下面的脚本将onnx模型转化为RV1126平台的rknn模型，

import os
import sys
import numpy as np
from rknn.api import RKNNONNX_MODEL = 'mask.onnx'
RKNN_MODEL = 'yolov5s_mask.rknn'if __name__ == '__main__':# 

这篇关于将onnx模型转化为RV1126平台的rknn模型的文章就介绍到这儿，希望我们推荐的文章对编程师们有所帮助！

---