睡觉状态的识别

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

一、数据爬取

从百度图片上爬取到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]]

posted @ 2020-03-16 08:00  ╄冷丶夜♂  阅读(576)  评论(0编辑  收藏  举报