A utility toolkit for converting RKNN2 models, potentially used during the model transformation process.
Project description
rknntools
- rknn模型转换可能会用到的工具
- RKNNTOOLKIT2版本 >= 1.5.2
代码说明
op_limit.py 包括函数check_op_limit，输入onnx模型，输出算子匹配情况
其他说明
- 目前算子已经匹配实例中的模型，以及conformer、zipformer等，后续可能有补充
- 目前依赖onnx库，测试1.13.1可行
TODO
- ONNX模型输入输出信息打印
- 读取ONNX模型并初始化全0输入，保存输出至文件供精度比对
- 修复ONNX模型的输入shape，满足rknntoolkits的要求
- 提供PT->ONNX代码，返回其中不支持算子在代码中的具体位置
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
rknntools-0.0.1.tar.gz
(3.9 kB
view details)
Built Distribution
Filter files by name, interpreter, ABI, and platform.
If you're not sure about the file name format, learn more about wheel file names.
Copy a direct link to the current filters
File details
Details for the file rknntools-0.0.1.tar.gz.
File metadata
- Download URL: rknntools-0.0.1.tar.gz
- Upload date:
- Size: 3.9 kB
- Tags: Source
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/4.0.2 CPython/3.9.13
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
f0bb377e13973889e006e29f6b2b709def15fddb7c6f32f775918c9bde0c830b
|
|
| MD5 |
6f02de7b11d8880c4127696347bd2b82
|
|
| BLAKE2b-256 |
e8fd80cfc9357e4a53f15f78a09f656297bbbbad9b8635f596e8e47987bda645
|
File details
Details for the file rknntools-0.0.1-py3-none-any.whl.
File metadata
- Download URL: rknntools-0.0.1-py3-none-any.whl
- Upload date:
- Size: 4.7 kB
- Tags: Python 3
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/4.0.2 CPython/3.9.13
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
b5c7cef7fc401ed159d0171e971dbf752b4f4183568f1187efd360d6488dc166
|
|
| MD5 |
090f74c2b6a81ac19f7be42f1ae93eb7
|
|
| BLAKE2b-256 |
0e8207cd7cf5a1262a01d1d1fc7892d87e8d84d51a66100f34cc69189c23b280
|
