本文核心内容:
1:介绍如何通过RK工具链将自己的模型转换量化为rknn模型
2:介绍模型转换过程中的注意事项
环境安装:
1:参考上一篇文章,此处不再重复
Rockchip实战1-跑通官方模型转换
正文部分:以RK1106为例
1:参考文档
A:rknn-toolkit2-master/doc/Rockchip_User_Guide_RKNN_Toolkit2_CN-1.4.0.pdf
直接上完整代码:
## -*- coding: utf-8 -*-
import os
import sys
import numpy as np
import cv2
import math
import onnx
import onnxruntime
from rknn.api import RKNN
INPUT_MODEL = 'models/faceMask_0_1_0_0.onnx' # 自己的onnx模型
OUTPUT_MODEL = '