(嵌入式AI)关于在stm32上做ai的一些问题

本文探讨了在STM32CubeIDE上进行嵌入式AI开发的问题,包括PyTorch模型转换限制,如不支持某些语句(如`img.shape[0]`),cubeAI 7.0的稳定性问题,建议使用6.0版本。验证数据需为二维CSV,输出用onehot编码,并介绍如何设置Flash和RAM资源。压缩仅针对全连接层,官方文档提供了详细指导。
摘要由CSDN通过智能技术生成

**(嵌入式AI)关于在stm32上做ai的一些问题,工具用的是STM32CubeIDE

也可以用STM32CubeMX,用IDE更好

1.pytorch中很多语句是不可以用cubeAI analyze的。
例如:feature.view(img.shape[0], -1)。。。img.shape[0]这个解析不了

2.cubeai7.0不稳定,会出现一些错误最好用6.0

3.Validate on desktop/target时选择csv文件只可以是二维数据,不可是三维,所以一般验证一batch(多维)的数据用npy格式文件,
注意的是:output(期望输出)用的是onehot编码的格式数据

4.可以格外设置可以加flash和ram(可以直接看下面的详细文档)

5.压缩只对全连接层压缩
这个官方的文档描述官方文档对compression的描述
也可以鼠标悬浮在AI位置,ctrl+d打开包内文档,里面会有说明可转换的操作等等东西。
在这里插入图片描述

使用手册的文档链接:(https://www.st

  • 3
    点赞
  • 11
    收藏
    觉得还不错? 一键收藏
  • 2
    评论
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值