请选择 进入手机版 | 继续访问电脑版

你的浏览器版本过低,可能导致网站不能正常访问!
为了你能正常使用网站功能,请使用这些浏览器。

Cube.AI【4】cifar10 在魔改Nucleo STMF767平台上的测试

[复制链接]
lebment 发布时间:2020-3-9 22:43
  希望论坛支持 一下markdown,我的很多程序readme都是markdown写的。  最近ST弄了个创意大赛,我参加了AI组,今天特意分享了我的测试流程。markdown写的,所以凑合看。希望各位参加比赛的顺利完成,我在疫区所以官方H7-DISCO板子收不到,等,有什么问题的话论坛私信我不能及时回复,可以在我别的帖子找我的联系方式。
/ F% b; r2 h; I( L/ i$ E2 [  测试记录,不详细描述怎么做的,一是我觉得手把手教别人怎么做,简直是害了别人,特别是学生,二是深度学习确实所需要知识点比较多,没办法展开。
; E# _- c9 ~' a7 t9 z  板子玩的开心就行!!!& M9 Z  T: {% r; Q

+ v7 A3 A9 n& {9 D) j# cifar10 在 nucleo STM32F767 平台的 测试
$ C0 w$ c9 P- ?( x3 z## 软件配置, P9 X, y% c- |; K! [( @
### win106 R& E/ w/ D0 _3 [/ K8 p3 n; `
   1. cubeMX 5.6.05 g7 ]% b3 M1 t0 j
   2. MDK 5.29a& o4 Q  f) d% L7 m
   3. cubemx扩展包--cube.AI 5.0.0
1 a) o; c# s6 |% N" V### ubuntu16.04 & J4 g" F/ m5 v" r
   1. tensorflow 2.1.0 (内含keras)
2 k" @- R6 X* x8 D4 I% Q   2. python 2.7 (ubuntu16.04 自带)
" O, D& O2 g3 o% D) Q9 t   3. cifar10 训练数据
' Z  _+ ]! E" Y: s5 t
  I/ m* @& B4 L' e2 t7 [6 s## 硬件配置   
7 v# d1 W; f' ?& a& B   1. nucleo F767
1 Z( |& f- z; l   2. 自制扩展板  R1 s: ]" V& S3 k# w$ }
   3. SD卡
/ V+ `4 H% S5 k$ U   4. 扩展板 连接 ST-Link RX TX 引脚的杜邦线
/ [/ _% ?6 Q8 \9 n2 Z3 M
2 E# E9 T: z7 p% I## 测试步骤
2 V. p+ G2 y* ~, V   * 在 ubuntu16.04 的操作不叙述( y1 r  G4 ~' V
   * 直接将转换的模型(模型约4MB),利用cubemx转换,暂称为STM32_AI_Model。
批注 2020-03-09 225751.png 批注 2020-03-09 225904.png
# u6 w- v$ s- d% F0 J
" @0 N! T5 x% k1 M' H   * 模型参数存储在W25Q128中,系统初始化时,QSPI进行内存映射,方便STM32_AI_Model读取模型参数。
5 K8 z4 h  n: F" D! {, C   * SD卡中读取29张测试图片均来自互联网,利用bmpconvST转换并缩小至32*32分辨率。
0 l& e. n, I5 T$ E2 Y+ X0 ~   * RGB每个通道需要进行预处理,即R/255 G/255 B/255
: i: b! M0 `! \/ ?3 V   * 图片在STM32_AI_Model的模型输入数据组织应为:' ]0 Y2 W3 c9 `5 |, V# D1 s2 X
| R1 | G1 | B1 | R2 | G2 | B2 | ... |
6 F* [9 b1 t& Z6 ?   * 循环从SD卡读取BMPt图片,显示在LTCD驱动的屏幕上,经过STM32_AI_Model推理后的结果,串口打印出来。7 H. p, A$ R; D3 b
推理结果的数据组织为:1 @$ f8 }5 F4 b( i
| C1 | C2 | C3 | C4 | ... | C10 |,每一哥数据代表对应种类的可能性。, K) A. `* @) P2 n6 m4 A# z
% p% u) J# ^  Z; }! y* _5 E
   * 可能性最大的即为推理最有可能的种类,cifar10即十种。
  O. C( Y, x. M; K5 l& R6 J7 ^6 W# {# M' F2 `
## 测试结果' [( T" c# o6 j8 f' G
  ### 串口打印:( Z: O2 h& `2 J9 S0 m% B4 ]
    sd_card_mount_ok- j& N/ G0 \- W+ p2 X0 J4 I
    airplane: 1.000       *4 a9 j% Q0 [- R" q7 g+ _
    airplane: 1.000       *1 b  b9 I6 w, l* b6 D# M) z
    automobile: 0.998     *4 D; f- G3 y2 N
    automobile: 1.000     *# V' O1 O! X, |  n
    airplane: 1.000       x
! H1 h' X/ [% v    dog: 0.584            x8 e/ d2 v0 w% L2 q( @# a
    bird: 1.000           *
! r; a4 N( k/ r$ M6 k, ]    dog: 0.999            x
- f" k5 ^/ m  _2 ?    dog: 0.874            x 6 h' v" F* [1 V$ k
    frog: 1.000           x
; Q2 G8 o* R: K    cat: 0.993            *
3 ~5 n  ^+ ?9 }    dog: 0.764            x
+ }* K$ d/ Q' G" l3 n    deer: 1.000           *
; p: W; |! ]' G8 L/ r6 b    airplane: 0.906       x. }/ {3 L( r' v8 i" W' M8 }: O
    dog: 1.000            *
" J2 R5 z: |+ [% S7 Q* U# r6 W9 p    dog: 0.999            *
" h; J2 K) @" b    deer: 0.810           x3 E8 D* Z( K$ v# r' D7 T
    bird: 1.000           x' r7 M6 [6 G& `' y' g) k; Q! Q
    frog: 0.688           *
7 m3 j$ Q- [6 s, M; O- e    frog: 1.000           *
7 ?; o& x7 o5 ?4 g    horse: 1.000          *
0 @( m1 J- Q% A( l4 c    horse: 1.000          *
# ^1 q, s# k" ]% n. o    horse: 1.000          *7 e) x( P- F, N! V4 g4 r( l
    ship: 0.664           *
3 v0 X5 Z! q- b6 ~  M    ship: 1.000           *
- P4 X# H1 K2 S/ p; W    ship: 1.000           *
, `9 }$ h$ _% \9 m4 x    truck: 1.000          *
% n3 y' {* ]2 n4 M+ T( A+ v$ r    ship: 0.658           x
- y) L1 \6 P- Q! N/ \    truck: 1.000          *
" t; e5 ], q# g: ~  H
+ b% d" k; t1 w![LCD截屏图片](./scr_print.png)
scr_print.png 我的LCD显示不正常,所以我用文件系统截了个图:
- o, K9 i) g; g% n2 t IMG_20200309_224522.jpg
9 o. L1 c2 {! S+ b% c& d* n
3 q' q  Q' \% V, @% C" u( ]### 测试准确率" G. k4 v$ [4 U* |
& l4 ~9 G1 C, k4 b3 J0 M" j
  * 19/29*100%=65.5%
5 K+ h7 a$ [0 Q6 e7 y6 ^
; a+ b1 f: B% t$ s# O### 测试总结6 i8 I: |$ t2 F: D0 U
  * 图片中比较常见的东西模型很容易辨认,图片第二行第三个是青蛙头上有一个蜗牛,辨认错了,还有四只猫居然只对了一只,其余的比较正常,毕竟图片分辨率不高。2 M4 G0 g" {5 j" P3 a" a' I
  * 由于我是用的深度学习模型是cifar10测试准确率比较高的,所以模型比较大,计算复杂,STM32F7推理时间约四五秒。# p2 s+ l- B: T9 m

3 E% {6 q, s  o- w. C0 b% H& t
# K+ I+ @- ^# R0 A0 F
' n- \' |, Y" y+ j0 h2 l6 E. \7 j& u2 ~# \* C7 g3 J, h4 c% _
8 }( H0 V& q) A: [! E2 x% E; e0 x, z

8 R" t8 \# H) M" _9 h5 e: K
4 i: V/ `  X' X' e- u* k- _- M# c

评分

参与人数 1 ST金币 +2 收起 理由
じ太阳当空照 + 2

查看全部评分

1 收藏 1 评论3 发布时间:2020-3-9 22:43

举报

3个回答
Yude 回答时间:2022-3-3 14:25:04

@楼主,只是想告知一下Markdown的编辑器已经就位,欢迎楼主输出更多关于AI方面的见解

lebment 回答时间:2022-6-11 18:27:03

Yude 发表于 2022-3-3 14:25
@楼主,只是想告知一下Markdown的编辑器已经就位,欢迎楼主输出更多关于AI方面的见解
...

两年了,很少逛论坛了

Yude 回答时间:2022-6-14 19:06:09
lebment 发表于 2022-6-11 18:274 b8 v& E. z; F% D0 ?
两年了,很少逛论坛了
* Q9 ]2 i0 {. W& T. j+ [. i3 c
哈哈, 欢迎多回来逛逛,提提宝贵意见; b7 N5 r2 F+ Y3 m3 e
关于意法半导体
我们是谁
投资者关系
意法半导体可持续发展举措
创新和工艺
招聘信息
联系我们
联系ST分支机构
寻找销售人员和分销渠道
社区
媒体中心
活动与培训
隐私策略
隐私策略
Cookies管理
行使您的权利
关注我们
st-img 微信公众号
st-img 手机版