4.1 Demo Zoo

简介

Bianbu AI Demo Zoo 是由进迭时空自主开发的示例项目，旨在为 K1 系列芯片提供多种深度学习模型的部署参考，实现端到端的推理流程演示。本项目提供 CV 和 NLP 两个分支，CV 分支涵盖图像分类、目标检测、人脸识别等典型任务，支持 C++ 和 Python 两种开发语言，适用于多种实际部署场景。

项目地址：⭐ Bianbu AI Demo Zoo | CV

支持的 CV 模型类型与具体列表详见 CV 模型列表。当前已覆盖常见的分类网络（ResNet、MobileNet）、检测网络（YOLOv5、YOLOX）、人脸识别（ArcFace）等模型。

示例 Demo

大多数模型均提供 Python 和 C++ 两种推理示例。用户在按照 README.md 下载必要的模型权重和测试数据后，只需几步即可完成推理演示。

Python Demo 示例

以 ResNet 图像分类为例：

cd python
python test_resnet.py  # 默认使用 ResNet50 模型

模型运行结束后将输出预测的类别标签。

C++ Demo 示例

cd cpp
mkdir build
cd build
cmake ..
make -j8
./resnet_demo --model /path/to/resnet50.onnx --image /path/to/image.jpg

运行结束后，同样会在终端输出预测结果，可结合 OpenCV 渲染分类标签。