STM32
直播中

萧蔼晨

8年用户 995经验值
私信 关注
[问答]

用STM32CubeMX导入神经网络,aiRun的indata应该定义成什么格式呢?

我用STM32CubeMX中的X-Cube-AI, 导入了一个处理时间序列的神经网络,网络input是2维数据,如(10,256,2)

在Generate code之后,在main.c文件中,有一个aiRun函数,需要输入一个in_data,这里的in_data我应该定义成什么格式呢?



回帖(1)

刘珊珊

2024-3-13 15:08:33
这个in_data是存放模型输入数据的buffer,数据存放的顺序跟你模型训练时的顺序是一样的,也就是(10, 256, 2)。所以buffer的总大小应该是10x256x2。如果是浮点模型就是float32,如果是8位有符号量化模型,就是int8。
具体的用法可以在X-CUBE-AI文档里看到代码示例。文档一般在此路径下:C:Usersuser_nameSTM32CubeRepositoryPacksSTMicroelectronicsX-CUBE-AI7.3.0Documentationembedded_client_api.html。其中user_name是计算机的用户名,7.3.0是X-CUBE-AI的版本号,根据安装版本的不同会有所不同。
举报

更多回帖

发帖
×
20
完善资料,
赚取积分