20221229:Rockchip实战2-跑通自有模型转换

本文核心内容:

1:介绍如何通过RK工具链将自己的模型转换量化为rknn模型

2:介绍模型转换过程中的注意事项

环境安装:

1:参考上一篇文章,此处不再重复

Rockchip实战1-跑通官方模型转换

正文部分:以RK1106为例

1:参考文档

A:rknn-toolkit2-master/doc/Rockchip_User_Guide_RKNN_Toolkit2_CN-1.4.0.pdf

20221229:Rockchip实战2-跑通自有模型转换_第1张图片

 直接上完整代码:

## -*- 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 = 'outputmodel

你可能感兴趣的:(AI-嵌入式-部署,边缘计算,iot,人工智能)