视频教程:
到此一游7758258的个人空间_哔哩哔哩_bilibili
搜024期
效果演示图如下:
完整的文件展示如下:
算法部分
其中dataset文件夹下存放的图像数据集,包括alert(注意力集中)、non_vigilant(漫不经心)、tired(疲劳)等。
运行01训练数据集文本生成.py会将数据集图片路径保存在txt文本中。
运行02CNN迁移学习训练模型.py会将txt文本中的图像数据读取进行模型的训练,最后保存在weights文件夹下。
运行03flask_server.py可打开接口连接小程序,接收小程序传来的图片,再将识别结果回传。
其他版本