4.1.2 RDK X3 Model Zoo 使用说明
分支与系统要求
RDK X3 使用 rdk_x3 分支,推荐系统版本 RDK OS >= 3.0.0 。
git clone https://github.com/D-Robotics/rdk_model_zoo.git
cd rdk_model_zoo
git checkout rdk_x3
仓库目录结构
rdk_x3 分支按 demo 目录组织,主要结构如下:
rdk_model_zoo/
|-- demos/
| |-- classification/ # 图像分类
| | |-- GoogLeNet/
| | |-- MobileNetV1/
| | |-- MobileNetV2/
| | |-- MobileNetV4/
| | |-- MobileOne/
| | |-- RepGhost/
| | |-- RepVGG/
| | |-- RepViT/
| | `-- ResNet/
| |-- detect/ # 目标检测
| | |-- FCOS/
| | |-- PaddleOCR/
| | |-- YOLOv10/
| | |-- YOLOv5/
| | `-- YOLOv8/
| `-- Instance_Segmentation/ # 实例分割
| `-- YOLOv8-Seg/
`-- resource/ # 文档资源
Python 推理接口
RDK X3 分支的 Python 推理接口有两种:
bpu_infer_lib_x3:需手动安装的 Python 推理库hobot_dnn.pyeasy_dnn:板端系统自带的 Python 推理接口
具体使用哪种接口,以目标 demo 目录 README 和源码入口为准。
警告
bpu_infer_lib_x3 和 hobot_dnn.pyeasy_dnn 对 featuremap 输入模型支持不佳。如需推理 featuremap 输入模型,建议使用 C++ 接口或升级到 RDK X5 平台并使用 hbm_runtime 推理接口。
安装 bpu_infer_lib_x3
RDK X3 使用 bpu_infer_lib_x3 的 demo,按以下命令安装:
wget -nc https://archive.d-robotics.cc/downloads/rdk_model_zoo/rdk_x3/bpu_infer_lib_x3-1.0.3-py3-none-any.whl
pip install bpu_infer_lib_x3-1.0.3-py3-none-any.whl
使用 hobot_dnn.pyeasy_dnn 的 demo 直接使用板端系统自带接口,无需额外安装。
快速上手
运行 demo
cd demos/<task>/<demo>
less README.md
# 按 README 准备模型、安装依赖并运行