巢湖学院学报 ›› 2021, Vol. 23 ›› Issue (6): 114-121.doi: 10.12152/j.issn.1672-2868.2021.06.016

• 工程与技术 • 上一篇    下一篇

基于RISC-V的口罩识别系统

张辉,周轶斐:安徽师范大学皖江学院 电子工程系   

  1. 安徽师范大学皖江学院 电子工程系,安徽 芜湖 241000
  • 收稿日期:2021-04-13 出版日期:2021-11-25 发布日期:2022-03-07
  • 作者简介:张辉(1984—),男,江苏泰兴人,安徽师范大学皖江学院电子工程系副教授,主要从事单片机与嵌入式研究。
  • 基金资助:
    安徽省高校自然科学研究项目(项目编号:KJ2020A1192)

Mask Recognition System Based on RISC-V

ZHANG Hui, ZHOU Yi-fei: Department of Electronic Engineering, Wanjiang College of Anhui Normal University   

  1. Department of Electronic Engineering, Wanjiang College of Anhui Normal University, Wuhu Anhui 241000
  • Received:2021-04-13 Online:2021-11-25 Published:2022-03-07

摘要: 基于RISC-V架构嵌入式平台,使用卷积神经网络算法实现了一种口罩识别系统。通过比较目前常用的深度学习算法,选择使用TensorFlow平台、YOLO算法以及口罩识别模型,并对其进行适量的稀疏训练,来减少模型大小,使其可以在嵌入式平台上运行,再利用Dlib训练特征点检测模型,对截取到的人脸进行提取并保存特征值,达到实现人脸检测的目的。实验以LFW数据集为实验样本,通过实验证明该方法可以达到较高准确率。

关键词: RISC-V架构, TensorFlow, YOLO, 人脸检测, 口罩识别

Abstract: This study uses convolutional neural network algorithm to implement a mask recognition system based on RISC-V architecture embedded platform. By comparing the current deep learning algorithms, we choose to use TensorFlow platform, YOLO algorithm and mask recognition model, and train them with appropriate amount of sparsity to reduce the model size so that they can run on the embedded platform. Then Dlib is adopted to train the feature point detection model to extract and save the feature values of the face, which can achieve the goal of recognizing faces. The LFW dataset is used as the experimental sample, and the experiment proves that the method can achieve high accuracy.

Key words: RISC-V architecture, TensorFlow, YOLO, face detection, mask recognition

中图分类号: 

  • TP391.41