Skip to main content

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代码,返回其中不支持算子在代码中的具体位置

Metadata

Release files for rknntools 0.0.3

For a detailed explanation of source distributions (sdists) and built distributions (wheels), please see the package formats documentation.

Source distribution (sdist)

Source distribution for rknntools 0.0.3
File Size Uploaded
rknntools-0.0.3.tar.gz 4.1 kB Details

Built distribution (wheel)

Table of built distributions (wheels) for rknntools 0.0.3
File Interpreter ABI Platform
rknntools-0.0.3-py3-none-any.whl Python 3 none any Details

Total release size: 9.0 kB

Release files / rknntools-0.0.3.tar.gz

Download URL rknntools-0.0.3.tar.gz
Size 4.1 kB
Tags Source
SHA-256 checksum
How to use checksums
c88b59b0468c5f97bcedebd71be034a2b8bccccc96214acc116a5bc16fefce40
BLAKE2b-256 checksum
How to use checksums
dd472d9a4d02ddb60f232fcc248b9ca4e0099c094ba525c563c00754ee8da30f
Upload date
Uploaded using Trusted Publishing?
What is trusted publishing?
No
Uploaded via twine/4.0.2 CPython/3.9.13

Release files / rknntools-0.0.3-py3-none-any.whl

Download URL rknntools-0.0.3-py3-none-any.whl
Size 5.0 kB
Tags Python 3
SHA-256 checksum
How to use checksums
b26529c8ba2ecb688b72d4f74d3864d1ad21fad39350438c13b84710b05625a1
BLAKE2b-256 checksum
How to use checksums
d6615be30092181cd22ea198cbe50c1d6e04958abfd2776320690c59a36bff33
Upload date
Uploaded using Trusted Publishing?
What is trusted publishing?
No
Uploaded via twine/4.0.2 CPython/3.9.13

Release history Release notifications | RSS feed

This release

0.0.3 This release

2 release files

0.0.2

2 release files

0.0.1

2 release files

Anthropic, PBC Visionary sponsor Bloomberg Visionary sponsor Hudson River Trading Visionary sponsor Meta Visionary sponsor NVIDIA Visionary sponsor Microsoft Sustainability sponsor Depot Continuous Integration AWS Cloud computing and Security Sponsor Datadog Monitoring Fastly CDN Google Download Analytics Sentry Error logging StatusPage Status page