睡觉状态的识别

Posted 2021-03-05 lzq666

本系统根据人体姿势判断图中人物是否在睡觉

一、数据爬取

从百度图片上爬取到1000张以上的人体非睡觉图片和睡觉姿势的图片

二、数据抽取

对每一张图片进行关键点检测，将数据存储至csv文件，并添加标记位0（没有睡觉）和1（睡觉）。由于决定睡觉姿态的关键点为

key_points = [‘top_head‘, ‘neck‘, ‘left_shoulder‘, ‘right_shoulder‘, ‘left_elbow‘, ‘right_elbow‘, ‘left_wrist‘, ‘right_wrist‘, ‘left_hip‘, ‘right_hip‘, ‘left_knee‘, ‘right_knee‘, ‘left_ankle‘, ‘right_ankle‘]，所以，只需获取这些关键点。

三、数据清理

清除含有大量空值的数据与重复数据

四、机器学习建模

使用sklearn随机森林回归模型进行建模

五、模型评估

混淆矩阵：

[[125 0]
[ 13 11]]