目标检测一般是yolov3、yolov4、yolov5、yolox、PSPnet、faster_rcnn、SDD等
视频教程如下,查025-028期:
025-028目标检测表情识别深度学习小程序网页视频_哔哩哔哩_bilibili
效果演示图如下:
完整的文件展示如下:
算法部分
其中data文件夹下存放的图像数据集,包括alert(注意力集中)、non_vigilant(漫不经心)、tired(疲劳)等。
运行01makeTxt.py会将数据集图片路径保存在txt文本中。
运行02train.py会将txt文本中的图像数据读取进行模型的训练,最后保存在runs文件夹下。
运行03detector_photo.py可以实现对单张图片的检测。
运行04server.py会生成一个接口供小程序交互,接受小程序传来的图片,再将识别好的图片传回去。