定制 AI 模型安装指南

用户4425

2024年12月31日修改

AI 模型转换并成功获得“。nb”文件后，请参考以下步骤应用到 Arduino SDK 中。​

重命名自定义的“。nb”文件

对象检测，“yolov3_tiny.nb”、“yolov4_tiny.nb”或“yolov7_tiny.nb”。​

人脸检测，“scrfd_500m_bnkps_640x640_u8.nb”。

人脸识别，“mobilefacenet_int16.nb”。

音频相关，“yamnet_fp16.nb”或“yamnet_s_hybrid.nb”。

将“。nb”文件复制/替换到项目文件夹中

1.对于 Windows 用户

•
C:\Users\ 用户名\AppData\Local\Arduino15\packages\realtek\hardware\AmebaPro2\版本\libraries\NeuralNetwork\examples​

•
或者用户自定义带有“。ino”文件的项目文件夹。​

2.对于 Linux 用户

•
\home\用户名\。arduino15\packages\realtek\hardware\AmebaPro2\版本 \libraries\NeuralNetwork\examples​

•
或者用户自定义带有“。ino”文件的项目文件夹。​

编译并运行示例

打开 Arduino IDE 并运行与传输的模型相关的 AmebaNN 示例。

具体选型请参考下面的示例指南和 API 文档。

•

•

◦
“神经网络”部分​

通过 SD 卡加载神经网络模型

导航到“文件”->“示例”->“AmebaNN”，打开 Ameba 神经网络示例之一。我们以 ObjectDetectionCallback 为例。​

在 Arduino IDE 上打开示例后，导航至“Tools”->“NN Model Load from:”，然后选择 SD 卡作为选项。​

接下来，在 modelSelect 函数下的代码中，只需将所使用的模型从“DEFAULT”更改为“CUSTOMIZED”即可。​

在 SD 卡中，创建一个名为“NN_MDL”的文件夹，并将模型保存在该文件夹中，如下所示命名模型。​

定制 AI 模型安装指南​