STM32F103嵌入式单片机运行CNN卷积神经网络图像分类识别

[视频作者] 枝江大学骨科学院

[视频时长] 2:7

[视频类型] 计算机技术

30岁开始学编程系列 第二期~ 硬件采用正点原子的stm32f103精英版与3.5TFTLCD。 模型采用经典的LENET5 CNN卷积神经网络模型,输入数据为32x32三通道彩色图片，训练集为CIFAR10, 神经网络模型在电脑端用pytorch训练后，提取权重参数至stm32。PC端通过串口传输与stm32通信，上位机发送图片像素点数据至stm32，stm32运行推理模型识别图片。 由于模型能力有限，且很多训练图片人眼也分辨不清，PC端训练验证集准确率仅为65%左右，最终导致使stm32推理准确性不高