rknn-toolkit2 2.1.0__tar.gz
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- rknn_toolkit2-2.1.0/LICENSE +5 -0
- rknn_toolkit2-2.1.0/PKG-INFO +99 -0
- rknn_toolkit2-2.1.0/README.md +84 -0
- rknn_toolkit2-2.1.0/pyproject.toml +21 -0
- rknn_toolkit2-2.1.0/setup.cfg +4 -0
- rknn_toolkit2-2.1.0/setup.py +18 -0
- rknn_toolkit2-2.1.0/src/rknn-toolkit2/__init__.py +0 -0
- rknn_toolkit2-2.1.0/src/rknn_toolkit2.egg-info/PKG-INFO +99 -0
- rknn_toolkit2-2.1.0/src/rknn_toolkit2.egg-info/SOURCES.txt +9 -0
- rknn_toolkit2-2.1.0/src/rknn_toolkit2.egg-info/dependency_links.txt +1 -0
- rknn_toolkit2-2.1.0/src/rknn_toolkit2.egg-info/top_level.txt +1 -0
|
@@ -0,0 +1,5 @@
|
|
|
1
|
+
**Copyright Statement**
|
|
2
|
+
|
|
3
|
+
Copyright(C) 2024 Rockchip Electronics Co., Ltd. All rights reserved.
|
|
4
|
+
|
|
5
|
+
BY OPENING OR USING THIS FILE, RECEIVER HEREBY ACKNOWLEDGES AND AGREES THAT THE SOFTWARE/FIRMWARE AND ITS DOCUMENTATIONS ("ROCKCHIP SOFTWARE") RECEIVED FROM ROCKCHIP ON AN "AS-IS" BASIS ONLY WITHOUT ANY AND ALL WARRANTIES, EITHER EXPRESS, IMPLIED OR STATUTORY, INCLUDING, WITHOUT LIMITATION, ANY WARRANTY OR CONDITION WITH RESPECT TO TITLE, MERCHANTABILITY, FITNESS FOR ANY PARTICULAR PURPOSE, OR NON-INFRINGEMENT. NEITHER DOES ROCKCHIP PROVIDE ANY WARRANTY WHATSOEVER WITH RESPECT TO ANY OPEN SOURCE TECHNOLOGIES, THIRD-PARTY TECHNOLOGIES OR ANY STANDARD TECHNOLOGIES WHICH MAY BE SUPPORTED BY, INCORPORATED IN, OR SUPPLIED WITH THE ROCKCHIP SOFTWARE. RECEIVER EXPRESSLY ACKNOWLEDGES THAT IT IS RECEIVER'S SOLE RESPONSIBILITY TO OBTAIN AND MAINTAIN ALL NECESSARY LICENSES AND RIGHTS FROM THEIR RESPECTIVE OWNERS TO USE ANY SUCH THIRD-PARTY TECHNOLOGIES OR ANY STANDARD TECHNOLOGIES. RECEIVER'S SOLE AND EXCLUSIVE REMEDY AND ROCKCHIP'S ENTIRE AND CUMULATIVE LIABILITY WITH RESPECT TO THE ROCKCHIP SOFTWARE RELEASED HEREUNDER WILL BE, AT ROCKCHIP 'S OPTION, TO REVISE OR REPLACE THE ROCKCHIP SOFTWARE AT ISSUE, OR REFUND ANY FEES OR CHARGE PAID BY RECEIVER TO ROCKCHIP FOR SUCH ROCKCHIP SOFTWARE AT ISSUE.
|
|
@@ -0,0 +1,99 @@
|
|
|
1
|
+
Metadata-Version: 2.1
|
|
2
|
+
Name: rknn-toolkit2
|
|
3
|
+
Version: 2.1.0
|
|
4
|
+
Summary: RKNN Toolkit
|
|
5
|
+
Home-page: https://github.com/airockchip/rknn-toolkit2
|
|
6
|
+
Author: ai@rock-chips.com
|
|
7
|
+
Author-email: airockchip <ai@rock-chips.com>
|
|
8
|
+
Project-URL: Homepage, https://github.com/airockchip/rknn-toolkit2
|
|
9
|
+
Project-URL: Issues, https://github.com/airockchip/rknn-toolkit2/issues
|
|
10
|
+
Classifier: Programming Language :: Python :: 3
|
|
11
|
+
Classifier: Operating System :: OS Independent
|
|
12
|
+
Requires-Python: >=3.6
|
|
13
|
+
Description-Content-Type: text/markdown
|
|
14
|
+
License-File: LICENSE
|
|
15
|
+
|
|
16
|
+
# Description
|
|
17
|
+
RKNN software stack can help users to quickly deploy AI models to Rockchip chips. The overall framework is as follows:
|
|
18
|
+
<center class="half">
|
|
19
|
+
<div style="background-color:#ffffff;">
|
|
20
|
+
<img src="res/framework.png" title="RKNN"/>
|
|
21
|
+
</center>
|
|
22
|
+
|
|
23
|
+
In order to use RKNPU, users need to first run the RKNN-Toolkit2 tool on the computer, convert the trained model into an RKNN format model, and then inference on the development board using the RKNN C API or Python API.
|
|
24
|
+
|
|
25
|
+
- RKNN-Toolkit2 is a software development kit for users to perform model conversion, inference and performance evaluation on PC and Rockchip NPU platforms.
|
|
26
|
+
|
|
27
|
+
- RKNN-Toolkit-Lite2 provides Python programming interfaces for Rockchip NPU platform to help users deploy RKNN models and accelerate the implementation of AI applications.
|
|
28
|
+
|
|
29
|
+
- RKNN Runtime provides C/C++ programming interfaces for Rockchip NPU platform to help users deploy RKNN models and accelerate the implementation of AI applications.
|
|
30
|
+
|
|
31
|
+
- RKNPU kernel driver is responsible for interacting with NPU hardware. It has been open source and can be found in the Rockchip kernel code.
|
|
32
|
+
|
|
33
|
+
# Support Platform
|
|
34
|
+
- RK3588 Series
|
|
35
|
+
- RK3576 Series
|
|
36
|
+
- RK3566/RK3568 Series
|
|
37
|
+
- RK3562 Series
|
|
38
|
+
- RV1103/RV1106
|
|
39
|
+
- RV1103B
|
|
40
|
+
- RK2118
|
|
41
|
+
|
|
42
|
+
|
|
43
|
+
Note:
|
|
44
|
+
|
|
45
|
+
**For RK1808/RV1109/RV1126/RK3399Pro, please refer to :**
|
|
46
|
+
|
|
47
|
+
https://github.com/airockchip/rknn-toolkit
|
|
48
|
+
|
|
49
|
+
https://github.com/airockchip/rknpu
|
|
50
|
+
|
|
51
|
+
https://github.com/airockchip/RK3399Pro_npu
|
|
52
|
+
|
|
53
|
+
|
|
54
|
+
# Download
|
|
55
|
+
- You can also download all packages, docker image, examples, docs and platform-tools from [RKNPU2_SDK](https://console.zbox.filez.com/l/I00fc3), fetch code: rknn
|
|
56
|
+
- You can get more examples from [rknn mode zoo](https://github.com/airockchip/rknn_model_zoo)
|
|
57
|
+
|
|
58
|
+
# Notes
|
|
59
|
+
- RKNN-Toolkit2 is not compatible with [RKNN-Toolkit](https://github.com/airockchip/rknn-toolkit)
|
|
60
|
+
- Currently only support on:
|
|
61
|
+
- Ubuntu 18.04 python 3.6/3.7
|
|
62
|
+
- Ubuntu 20.04 python 3.8/3.9
|
|
63
|
+
- Ubuntu 22.04 python 3.10/3.11
|
|
64
|
+
- Latest version:v2.1.0
|
|
65
|
+
|
|
66
|
+
|
|
67
|
+
|
|
68
|
+
# RKNN LLM
|
|
69
|
+
|
|
70
|
+
If you want to deploy LLM (Large Language Model), we have introduced a new SDK called RKNN-LLM. For details, please refer to:
|
|
71
|
+
|
|
72
|
+
https://github.com/airockchip/rknn-llm
|
|
73
|
+
|
|
74
|
+
|
|
75
|
+
|
|
76
|
+
# CHANGELOG
|
|
77
|
+
|
|
78
|
+
## v2.1.0
|
|
79
|
+
- Support RV1103B (Beta)
|
|
80
|
+
- Support RK2118 (Beta)
|
|
81
|
+
- Support Flash Attention (Only RK3562 and RK3576)
|
|
82
|
+
- Improve MatMul API
|
|
83
|
+
- Improve support for int32 and int64
|
|
84
|
+
- Support more operators and operator fusion
|
|
85
|
+
|
|
86
|
+
for older version, please refer [CHANGELOG](CHANGELOG.md)
|
|
87
|
+
|
|
88
|
+
# Feedback and Community Support
|
|
89
|
+
- [Redmine](https://redmine.rock-chips.com) (**Feedback recommended, Please consult our sales or FAE for the redmine account**)
|
|
90
|
+
- QQ Group Chat: 1025468710 (full, please join group 3)
|
|
91
|
+
- QQ Group Chat2: 547021958 (full, please join group 3)
|
|
92
|
+
- QQ Group Chat3: 469385426
|
|
93
|
+
<center class="half">
|
|
94
|
+
<img width="200" height="200" src="res/QQGroupQRCode.png" title="QQ Group Chat"/>
|
|
95
|
+
<img width="200" height="200" src="res/QQGroup2QRCode.png" title="QQ Group Chat2"/>
|
|
96
|
+
<img width="200" height="200" src="res/QQGroup3QRCode.png" title="QQ Group Chat3"/>
|
|
97
|
+
</center>
|
|
98
|
+
|
|
99
|
+
|
|
@@ -0,0 +1,84 @@
|
|
|
1
|
+
# Description
|
|
2
|
+
RKNN software stack can help users to quickly deploy AI models to Rockchip chips. The overall framework is as follows:
|
|
3
|
+
<center class="half">
|
|
4
|
+
<div style="background-color:#ffffff;">
|
|
5
|
+
<img src="res/framework.png" title="RKNN"/>
|
|
6
|
+
</center>
|
|
7
|
+
|
|
8
|
+
In order to use RKNPU, users need to first run the RKNN-Toolkit2 tool on the computer, convert the trained model into an RKNN format model, and then inference on the development board using the RKNN C API or Python API.
|
|
9
|
+
|
|
10
|
+
- RKNN-Toolkit2 is a software development kit for users to perform model conversion, inference and performance evaluation on PC and Rockchip NPU platforms.
|
|
11
|
+
|
|
12
|
+
- RKNN-Toolkit-Lite2 provides Python programming interfaces for Rockchip NPU platform to help users deploy RKNN models and accelerate the implementation of AI applications.
|
|
13
|
+
|
|
14
|
+
- RKNN Runtime provides C/C++ programming interfaces for Rockchip NPU platform to help users deploy RKNN models and accelerate the implementation of AI applications.
|
|
15
|
+
|
|
16
|
+
- RKNPU kernel driver is responsible for interacting with NPU hardware. It has been open source and can be found in the Rockchip kernel code.
|
|
17
|
+
|
|
18
|
+
# Support Platform
|
|
19
|
+
- RK3588 Series
|
|
20
|
+
- RK3576 Series
|
|
21
|
+
- RK3566/RK3568 Series
|
|
22
|
+
- RK3562 Series
|
|
23
|
+
- RV1103/RV1106
|
|
24
|
+
- RV1103B
|
|
25
|
+
- RK2118
|
|
26
|
+
|
|
27
|
+
|
|
28
|
+
Note:
|
|
29
|
+
|
|
30
|
+
**For RK1808/RV1109/RV1126/RK3399Pro, please refer to :**
|
|
31
|
+
|
|
32
|
+
https://github.com/airockchip/rknn-toolkit
|
|
33
|
+
|
|
34
|
+
https://github.com/airockchip/rknpu
|
|
35
|
+
|
|
36
|
+
https://github.com/airockchip/RK3399Pro_npu
|
|
37
|
+
|
|
38
|
+
|
|
39
|
+
# Download
|
|
40
|
+
- You can also download all packages, docker image, examples, docs and platform-tools from [RKNPU2_SDK](https://console.zbox.filez.com/l/I00fc3), fetch code: rknn
|
|
41
|
+
- You can get more examples from [rknn mode zoo](https://github.com/airockchip/rknn_model_zoo)
|
|
42
|
+
|
|
43
|
+
# Notes
|
|
44
|
+
- RKNN-Toolkit2 is not compatible with [RKNN-Toolkit](https://github.com/airockchip/rknn-toolkit)
|
|
45
|
+
- Currently only support on:
|
|
46
|
+
- Ubuntu 18.04 python 3.6/3.7
|
|
47
|
+
- Ubuntu 20.04 python 3.8/3.9
|
|
48
|
+
- Ubuntu 22.04 python 3.10/3.11
|
|
49
|
+
- Latest version:v2.1.0
|
|
50
|
+
|
|
51
|
+
|
|
52
|
+
|
|
53
|
+
# RKNN LLM
|
|
54
|
+
|
|
55
|
+
If you want to deploy LLM (Large Language Model), we have introduced a new SDK called RKNN-LLM. For details, please refer to:
|
|
56
|
+
|
|
57
|
+
https://github.com/airockchip/rknn-llm
|
|
58
|
+
|
|
59
|
+
|
|
60
|
+
|
|
61
|
+
# CHANGELOG
|
|
62
|
+
|
|
63
|
+
## v2.1.0
|
|
64
|
+
- Support RV1103B (Beta)
|
|
65
|
+
- Support RK2118 (Beta)
|
|
66
|
+
- Support Flash Attention (Only RK3562 and RK3576)
|
|
67
|
+
- Improve MatMul API
|
|
68
|
+
- Improve support for int32 and int64
|
|
69
|
+
- Support more operators and operator fusion
|
|
70
|
+
|
|
71
|
+
for older version, please refer [CHANGELOG](CHANGELOG.md)
|
|
72
|
+
|
|
73
|
+
# Feedback and Community Support
|
|
74
|
+
- [Redmine](https://redmine.rock-chips.com) (**Feedback recommended, Please consult our sales or FAE for the redmine account**)
|
|
75
|
+
- QQ Group Chat: 1025468710 (full, please join group 3)
|
|
76
|
+
- QQ Group Chat2: 547021958 (full, please join group 3)
|
|
77
|
+
- QQ Group Chat3: 469385426
|
|
78
|
+
<center class="half">
|
|
79
|
+
<img width="200" height="200" src="res/QQGroupQRCode.png" title="QQ Group Chat"/>
|
|
80
|
+
<img width="200" height="200" src="res/QQGroup2QRCode.png" title="QQ Group Chat2"/>
|
|
81
|
+
<img width="200" height="200" src="res/QQGroup3QRCode.png" title="QQ Group Chat3"/>
|
|
82
|
+
</center>
|
|
83
|
+
|
|
84
|
+
|
|
@@ -0,0 +1,21 @@
|
|
|
1
|
+
[build-system]
|
|
2
|
+
requires = ["setuptools>=50.0"]
|
|
3
|
+
build-backend = "setuptools.build_meta"
|
|
4
|
+
|
|
5
|
+
[project]
|
|
6
|
+
name = "rknn-toolkit2"
|
|
7
|
+
version = "2.1.0"
|
|
8
|
+
authors = [
|
|
9
|
+
{ name="airockchip", email="ai@rock-chips.com" },
|
|
10
|
+
]
|
|
11
|
+
description = "RKNN Toolkit"
|
|
12
|
+
readme = "README.md"
|
|
13
|
+
requires-python = ">=3.6"
|
|
14
|
+
classifiers = [
|
|
15
|
+
"Programming Language :: Python :: 3",
|
|
16
|
+
'Operating System :: OS Independent',
|
|
17
|
+
]
|
|
18
|
+
|
|
19
|
+
[project.urls]
|
|
20
|
+
Homepage = "https://github.com/airockchip/rknn-toolkit2"
|
|
21
|
+
Issues = "https://github.com/airockchip/rknn-toolkit2/issues"
|
|
@@ -0,0 +1,18 @@
|
|
|
1
|
+
import setuptools
|
|
2
|
+
|
|
3
|
+
with open("README.md", "r", encoding="utf-8") as fh:
|
|
4
|
+
long_description = fh.read()
|
|
5
|
+
|
|
6
|
+
setuptools.setup(
|
|
7
|
+
name="rknn-toolkit2",
|
|
8
|
+
version="2.1.0",
|
|
9
|
+
author="ai@rock-chips.com",
|
|
10
|
+
url='https://github.com/airockchip/rknn-toolkit2',
|
|
11
|
+
author_email="ai@rock-chips.com",
|
|
12
|
+
description="RKNN-TOOLKIT2 ",
|
|
13
|
+
long_description=long_description,
|
|
14
|
+
long_description_content_type="text/markdown",
|
|
15
|
+
package_dir={"": "src"},
|
|
16
|
+
packages=setuptools.find_packages(where="src"),
|
|
17
|
+
python_requires=">=3.6",
|
|
18
|
+
)
|
|
File without changes
|
|
@@ -0,0 +1,99 @@
|
|
|
1
|
+
Metadata-Version: 2.1
|
|
2
|
+
Name: rknn-toolkit2
|
|
3
|
+
Version: 2.1.0
|
|
4
|
+
Summary: RKNN Toolkit
|
|
5
|
+
Home-page: https://github.com/airockchip/rknn-toolkit2
|
|
6
|
+
Author: ai@rock-chips.com
|
|
7
|
+
Author-email: airockchip <ai@rock-chips.com>
|
|
8
|
+
Project-URL: Homepage, https://github.com/airockchip/rknn-toolkit2
|
|
9
|
+
Project-URL: Issues, https://github.com/airockchip/rknn-toolkit2/issues
|
|
10
|
+
Classifier: Programming Language :: Python :: 3
|
|
11
|
+
Classifier: Operating System :: OS Independent
|
|
12
|
+
Requires-Python: >=3.6
|
|
13
|
+
Description-Content-Type: text/markdown
|
|
14
|
+
License-File: LICENSE
|
|
15
|
+
|
|
16
|
+
# Description
|
|
17
|
+
RKNN software stack can help users to quickly deploy AI models to Rockchip chips. The overall framework is as follows:
|
|
18
|
+
<center class="half">
|
|
19
|
+
<div style="background-color:#ffffff;">
|
|
20
|
+
<img src="res/framework.png" title="RKNN"/>
|
|
21
|
+
</center>
|
|
22
|
+
|
|
23
|
+
In order to use RKNPU, users need to first run the RKNN-Toolkit2 tool on the computer, convert the trained model into an RKNN format model, and then inference on the development board using the RKNN C API or Python API.
|
|
24
|
+
|
|
25
|
+
- RKNN-Toolkit2 is a software development kit for users to perform model conversion, inference and performance evaluation on PC and Rockchip NPU platforms.
|
|
26
|
+
|
|
27
|
+
- RKNN-Toolkit-Lite2 provides Python programming interfaces for Rockchip NPU platform to help users deploy RKNN models and accelerate the implementation of AI applications.
|
|
28
|
+
|
|
29
|
+
- RKNN Runtime provides C/C++ programming interfaces for Rockchip NPU platform to help users deploy RKNN models and accelerate the implementation of AI applications.
|
|
30
|
+
|
|
31
|
+
- RKNPU kernel driver is responsible for interacting with NPU hardware. It has been open source and can be found in the Rockchip kernel code.
|
|
32
|
+
|
|
33
|
+
# Support Platform
|
|
34
|
+
- RK3588 Series
|
|
35
|
+
- RK3576 Series
|
|
36
|
+
- RK3566/RK3568 Series
|
|
37
|
+
- RK3562 Series
|
|
38
|
+
- RV1103/RV1106
|
|
39
|
+
- RV1103B
|
|
40
|
+
- RK2118
|
|
41
|
+
|
|
42
|
+
|
|
43
|
+
Note:
|
|
44
|
+
|
|
45
|
+
**For RK1808/RV1109/RV1126/RK3399Pro, please refer to :**
|
|
46
|
+
|
|
47
|
+
https://github.com/airockchip/rknn-toolkit
|
|
48
|
+
|
|
49
|
+
https://github.com/airockchip/rknpu
|
|
50
|
+
|
|
51
|
+
https://github.com/airockchip/RK3399Pro_npu
|
|
52
|
+
|
|
53
|
+
|
|
54
|
+
# Download
|
|
55
|
+
- You can also download all packages, docker image, examples, docs and platform-tools from [RKNPU2_SDK](https://console.zbox.filez.com/l/I00fc3), fetch code: rknn
|
|
56
|
+
- You can get more examples from [rknn mode zoo](https://github.com/airockchip/rknn_model_zoo)
|
|
57
|
+
|
|
58
|
+
# Notes
|
|
59
|
+
- RKNN-Toolkit2 is not compatible with [RKNN-Toolkit](https://github.com/airockchip/rknn-toolkit)
|
|
60
|
+
- Currently only support on:
|
|
61
|
+
- Ubuntu 18.04 python 3.6/3.7
|
|
62
|
+
- Ubuntu 20.04 python 3.8/3.9
|
|
63
|
+
- Ubuntu 22.04 python 3.10/3.11
|
|
64
|
+
- Latest version:v2.1.0
|
|
65
|
+
|
|
66
|
+
|
|
67
|
+
|
|
68
|
+
# RKNN LLM
|
|
69
|
+
|
|
70
|
+
If you want to deploy LLM (Large Language Model), we have introduced a new SDK called RKNN-LLM. For details, please refer to:
|
|
71
|
+
|
|
72
|
+
https://github.com/airockchip/rknn-llm
|
|
73
|
+
|
|
74
|
+
|
|
75
|
+
|
|
76
|
+
# CHANGELOG
|
|
77
|
+
|
|
78
|
+
## v2.1.0
|
|
79
|
+
- Support RV1103B (Beta)
|
|
80
|
+
- Support RK2118 (Beta)
|
|
81
|
+
- Support Flash Attention (Only RK3562 and RK3576)
|
|
82
|
+
- Improve MatMul API
|
|
83
|
+
- Improve support for int32 and int64
|
|
84
|
+
- Support more operators and operator fusion
|
|
85
|
+
|
|
86
|
+
for older version, please refer [CHANGELOG](CHANGELOG.md)
|
|
87
|
+
|
|
88
|
+
# Feedback and Community Support
|
|
89
|
+
- [Redmine](https://redmine.rock-chips.com) (**Feedback recommended, Please consult our sales or FAE for the redmine account**)
|
|
90
|
+
- QQ Group Chat: 1025468710 (full, please join group 3)
|
|
91
|
+
- QQ Group Chat2: 547021958 (full, please join group 3)
|
|
92
|
+
- QQ Group Chat3: 469385426
|
|
93
|
+
<center class="half">
|
|
94
|
+
<img width="200" height="200" src="res/QQGroupQRCode.png" title="QQ Group Chat"/>
|
|
95
|
+
<img width="200" height="200" src="res/QQGroup2QRCode.png" title="QQ Group Chat2"/>
|
|
96
|
+
<img width="200" height="200" src="res/QQGroup3QRCode.png" title="QQ Group Chat3"/>
|
|
97
|
+
</center>
|
|
98
|
+
|
|
99
|
+
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
rknn-toolkit2
|