Files
FastDeploy/examples/vision/detection/rkyolo/cpp/README_CN.md
Zheng-Bicheng 109d1046ae [Model] add function for setting anchor rknpu2 (#1728)
* add function for setting anchor rknpu2
add more demo for rknpu2
fixed md error

* Update config.h

---------

Co-authored-by: DefTruth <31974251+DefTruth@users.noreply.github.com>
2023-04-04 20:33:06 +08:00

1.5 KiB
Raw Blame History

English | 简体中文

RKYOLO C++部署示例

本目录下提供infer_xxxxx.cc快速完成RKYOLO模型在Rockchip板子上上通过二代NPU加速部署的示例。

在部署前,需确认以下两个步骤:

  1. 软硬件环境满足要求
  2. 根据开发环境下载预编译部署库或者从头编译FastDeploy仓库

以上步骤请参考RK2代NPU部署库编译实现

wget https://gitee.com/paddlepaddle/PaddleDetection/raw/release/2.4/demo/000000014439.jpg
cd build
cmake .. -DFASTDEPLOY_INSTALL_DIR=${PWD}/fastdeploy-linux-x64-x.x.x
make -j8

wget https://bj.bcebos.com/paddlehub/fastdeploy/rknpu2/yolov5-s-relu.zip
unzip yolov5-s-relu.zip
./infer_rkyolov5 yolov5-s-relu/yolov5s_relu_tk2_RK3588_i8.rknn 000000014439.jpg

wget https://bj.bcebos.com/paddlehub/fastdeploy/rknpu2/yolov7-tiny.zip
unzip yolov7-tiny.zip
./infer_rkyolov7 yolov7-tiny/yolov7-tiny_tk2_RK3588_i8.rknn 000000014439.jpg

wget https://bj.bcebos.com/paddlehub/fastdeploy/rknpu2/yolox-s.zip
unzip yolox-s.zip
./infer_rkyolox yolox-s/yoloxs_tk2_RK3588_i8.rknn 000000014439.jpg

常见问题

如果你使用自己训练的YOLOv5模型你可能会碰到运行FastDeploy的demo后出现segmentation fault的问题很大概率是label数目不一致你可以使用以下方案来解决:

model.GetPostprocessor().SetClassNum(3);