虽然没有找到hhb的官方示例,但是我找到了nncase的,在src/big/nncase/examples中
一、环境搭建
examples也有个readme,不过里面的环境搭建跟sdk中的有点差别,不过大差不差,docker容器已经启动了,需要在容器中安装nncase
cd /root/k230/k230_sdk-main/src/big/nncase/
pip install x86_64/*.whl
pip install nncase-kpu==2.2.0.20230728
虽然x86_64中提供的whl是2.1版本的,但是安装会自动下载2.2版本的安装
二、编译模型
移动到examples下
生成kmodel模型
python3 ./scripts/yolov5s_onnx.py --target k230 --model models/yolov5s.onnx --dataset calibration_dataset
验证生成的kmodel模型精度
export PATH=$PATH:/usr/local/lib/python3.8/dist-packages/
python3 scripts/yolov5s_onnx_simu.py --model models/yolov5s.onnx --model_input object_detect/data/input_fp32.bin --kmodel tmp/yolov5s_onnx/test.kmodel --kmodel_input object_detect/data/input_uint8.bin