如何套用你的客制化模型

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”文件复制/替换到项目资料夹中

对于 Windows 用户,
• C:\Users\USERNAME\AppData\Local\Arduino15\packages\realtek\hardware\AmebaPro2\VERSION\libraries\NeuralNetwork\examples
• 或带有“.ino”文件的用户自定义项目资料夹。

对于 Linux 用户,
• \home\USERNAME\.arduino15\packages\realtek\hardware\AmebaPro2\VERSION \libraries\NeuralNetwork\examples
• 或带有“.ino”文件的用户自定义项目资料夹。

编译并执行范例

打开 Arduino IDE 并运行相关的 AmebaNN 范例。

具体模型选择请参考相关范例API 文件

透过SD卡载入神经网路模型

至 “File” -> “Examples” -> “AmebaNN” 开启 Ameba 神经网路范例。 我们以 ObjectDetectionCallback 为例。
在 Arduino IDE 上开启范例后,导览至“Tools”->“NN Model Load from:”,然后选择 SD 卡作为选项。

get- start-1

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

get- start-1

在 SD 卡中,建立一个名为 “NN_MDL” 的资料夹。

get- start-1

并按如下所述命名模型,将模型保存在资料夹中,

Object Detection:
yolov3_tiny.nb
yolov4_tiny.nb
yolov7_tiny.nb
Face Detection:
scrfd.nb
Face Recognition:
mobilefacenet.nb
Audio Classification:
yamnet.nb
Image Classification:
imgclassification.nb

get- start-1

可以至 \Arduino15\packages\realtek\hardware\AmebaPro2\{SDK Version}\libraries\NeuralNetwork\src 来更改 SD_Model.cpp 中的 .nb 档案名,以符合 SD 卡上的.nb 档案名称。

get- start-1

请先确认已安装QQ通讯软体