This is an old revision of the document!
The SDK only supports python3.6
or python3.8
, here is an example of creating a virtual environment for python3.8
.
Follow this docs to install conda.
Then create a virtual environment.
conda create -n npu-env python=3.8 conda activate npu-env #activate conda deactivate #deactivate
After creating the virtual environment, start the virtual environment and start the next step.
Clone SDK from Rockchip Github.
git clone https://github.com/rockchip-linux/rknn-toolkit2.git
Install dependences and RKNN toolkit2 packages,
cd rknn-toolkit2 sudo apt-get install python3 python3-dev python3-pip sudo apt-get install libxslt1-dev zlib1g-dev libglib2.0 libsm6 libgl1-mesa-glx libprotobuf-dev gcc pip3 install -r doc/requirements_cp38-*.txt pip3 install packages/rknn_toolkit2-*-cp38-cp38-linux_x86_64.whl
Convert yolov5
as a example,
cd examples/onnx/yolov5/ python3 test.py
After convert, it will generate the rknn file yolov5s.rknn
.
$ ls
bus.jpg dataset.txt onnx_yolov5_0.npy onnx_yolov5_1.npy onnx_yolov5_2.npy test.py yolov5s.onnx yolov5s.rknn
For more usage, please refer to the related documents under doc
.