睡觉状态的识别
本系统根据人体姿势判断图中人物是否在睡觉
一、数据爬取
从百度图片上爬取到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]]