rknn环境安装笔记
目录
安装python环境:
conda config --set report_errors true
conda create --name rknn3.6 python=3.6
rknn下载:
http://repo.rock-chips.com/pypi/simple/rknn-toolkit/
rknn-toolkit-1.2.1-cp36-cp36m-linux_aarch64.whl 17-Oct-2019 10:05 56659109
rknn-toolkit-1.2.1-cp36-cp36m-linux_x86_64.whl 17-Oct-2019 10:05 60841540
rknn-toolkit-1.2.1-cp36-cp36m-macosx_10_9_x86_6..> 17-Oct-2019 10:08 36891121
rknn-toolkit-1.2.1-cp36-cp36m-win_amd64.whl 17-Oct-2019 10:08 33134089
rknn-toolkit-1.2.1-cp37-cp37m-linux_aarch64.whl 17-Oct-2019 10:09 57598878
rknn-toolkit-1.3.0-cp36-cp36m-linux_aarch64.whl 20-Jan-2020 06:47 57274013
rknn-toolkit-1.3.0-cp36-cp36m-linux_x86_64.whl 20-Jan-2020 06:59 136707878
rknn-toolkit-1.3.0-cp36-cp36m-macosx_10_9_x86_6..> 20-Jan-2020 07:02 116558624
rknn-toolkit-1.3.0-cp36-cp36m-win_amd64.whl 20-Jan-2020 07:06 109971468
rknn-toolkit-1.3.0-cp37-cp37m-linux_aarch64.whl 20-Jan-2020 07:06 58575235
rknn-toolkit-1.3.2-cp37-cp37m-linux_aarch64.whl 23-Apr-2020 06:48 35742584
rknn-toolkit-1.4.0-cp36-cp36m-linux_x86_64.whl 24-Aug-2020 00:51 135521194
rknn-toolkit-1.4.0-cp36-cp36m-win_amd64.whl 24-Aug-2020 00:51 108328413
rknn-toolkit-1.4.0-cp37-cp37m-linux_aarch64.whl 24-Aug-2020 00:51 31001197
rknn-toolkit-1.4.0-cp37-cp37m-macosx_10_15_x86_..> 24-Aug-2020 00:51 108429118
rknn_toolkit-1.6.0-cp35-cp35m-linux_aarch64.whl 14-Jan-2021 02:33 34538213
rknn_toolkit-1.6.0-cp35-cp35m-linux_x86_64.whl 14-Jan-2021 02:34 160362809
rknn_toolkit-1.6.0-cp36-cp36m-linux_x86_64.whl 14-Jan-2021 02:35 160552824
rknn_toolkit-1.6.0-cp36-cp36m-macosx_10_15_x86_..> 14-Jan-2021 02:36 119510062
rknn_toolkit-1.6.0-cp36-cp36m-win_amd64.whl 14-Jan-2021 02:36 112325076
rknn_toolkit-1.6.0-cp37-cp37m-linux_aarch64.whl 14-Jan-2021 02:36 39105410
rknn_toolkit-1.6.0-cp37-cp37m-macosx_10_15_x86_..> 14-Jan-2021 02:37 117990241
rknn_toolkit-1.7.0-cp37-cp37m-linux_aarch64.whl 02-Nov-2021 08:31 40410162
rknn_toolkit-1.7.1-cp35-cp35m-linux_aarch64.whl 11-May-2022 04:33 37938676
rknn_toolkit-1.7.1-cp35-cp35m-linux_x86_64.whl 11-May-2022 04:33 142780939
rknn_toolkit-1.7.1-cp36-cp36m-linux_x86_64.whl 11-May-2022 04:34 140752080
rknn_toolkit-1.7.1-cp36-cp36m-macosx_10_15_x86_..> 11-May-2022 04:34 122255122
rknn_toolkit-1.7.1-cp36-cp36m-win_amd64.whl 11-May-2022 04:34 114232401
rknn_toolkit-1.7.1-cp37-cp37m-linux_aarch64.whl 11-May-2022 04:34 41073863
rknn_toolkit-1.7.1-cp37-cp37m-macosx_10_15_x86_..> 11-May-2022 04:35 120252888
2022.06.18更新:
rknn windows版本,只有python3.6版本,linux有3.7版本
rknn_toolkit-1.7.1-cp36-cp36m-win_amd64.whl
yolox onnx导出rknn代码
-
import os
-
import urllib
-
import traceback
-
import time
-
import sys
-
import re
-
import numpy as np
-
import cv2
-
from yolox.data.data_augment import preproc as preprocess
-
from rknn.api import RKNN
-
import time
-
from yolox.utils import mkdir, multiclass_nms, demo_postprocess, vis
-
from config import Config
-
-
def run():
-
cfg = Config()
-
ONNX_MODEL = 'yolox.onnx'
-
RKNN_MODEL = 'yolox.rknn'
-
-
rknn = RKNN(verbose=True)
-
rknn.list_devices()
-
# rknn.config(target_platform='rk3566')
-
rknn.config()
-
ret = rknn.load_onnx(model=ONNX_MODEL)
-
if ret < 0:
-
print("load faild")
-
ret = rknn.build(do_quantization=False,dataset="/home/bona/Projects/python/YOLOX-main/demo/RKNN/dataset.txt")
-
if ret < 0:
-
print("build faild")
-
ret = rknn.export_rknn(RKNN_MODEL)
-
if ret < 0:
-
print("export faild")
-
print("export down")
文章来源: blog.csdn.net,作者:AI视觉网奇,版权归原作者所有,如需转载,请联系作者。
原文链接:blog.csdn.net/jacke121/article/details/125345883
- 点赞
- 收藏
- 关注作者
评论(0)