[发明专利]基于关节点坐标的人体姿势识别方法及系统

权利要求书

1.一种基于关节点坐标的人体姿势识别方法，其特征在于，所述方法包括：

启动外部应用模块；

所述外部应用模块读取人体姿势识别模块所需的配置参数信息，调用所述人体姿势识别模块的初始化接口并向所述人体姿势识别模块中传入所读取的配置参数信息；

所述人体姿势识别模块根据所述配置参数信息对待识别图像中的人体姿势进行识别；

所述外部应用模块间隔预设时间循环调用所述人体姿势识别模块的获取姿势识别结果接口，获取对应的人体姿势识别结果；其中，

所述人体姿势识别模块所需的配置参数包括：Yolo模型类型、Yolo模型文件路径、姿势模型类型、姿势模型文件路径、视频流url、视频流帧缩放后的高、视频流帧缩放后的宽、所使用GPU/CPU的设备号、可视区域比例、丢帧开关、丢帧比例、人体位置校准开关、人体位置校准区域坐标、人体位置校准区域展示开关、录制开关、录制路径、帧保存路径、姿势识别算法模型、姿势识别算法模型文件路径；所述人体姿势识别模块包括人体关节点坐标检测模块和调试子模块；

所述人体关节点坐标检测模块的配置参数包括：Yolo模型类型、Yolo模型文件路径、姿势模型类型、姿势模型文件路径、视频流url、视频流帧缩放后的高、视频流帧缩放后的宽、所使用GPU/CPU的设备号、丢帧开关、丢帧比例、人体位置校准开关、人体位置校准区域坐标、人体位置校准区域展示开关；其中，

所述人体关节点坐标检测模块用于执行循环工作的三个线程，所述三个线程分别为：视频流获取和解析服务线程、yolo服务线程、姿势服务线程；

所述人体关节点坐标检测模块的工作流程包括：

通过所述视频流获取和解析服务线程判断当前缓存及所述姿势服务线程状态，若缓存未满且所述姿势服务线程为已完成工作状态，则读取待识别的视频流的当前帧；

当所述当前帧读取完成时，判断丢帧开关状态，若丢帧开关状态为开启，则按丢帧比例进行丢帧处理；

当所述丢帧处理完成时，根据所述视频流帧缩放后的高和视频流帧缩放后的宽对当前帧进行缩放处理，生成处理结果；

将所述处理结果保存并通知所述yolo服务线程；

所述yolo服务线程调用yolo底层算法接口计算出所述当前帧的人像框；

根据校准框使用算法从所述人像框中筛选出最优人像框并获取处理结果；

将当前人像框的处理结果保存并通知所述姿势服务线程；

所述姿势服务线程调用姿势底层算法接口计算出人体姿势节点坐标；

若所述姿势节点坐标符合预设条件，则通过回调函数将所述姿势节点坐标结果传给所述人体姿势识别模块；

将上述处理结果保存并通知视频流获取和解析服务线程，并循环执行上述步骤。

2.根据权利要求1所述的基于关节点坐标的人体姿势识别方法，其特征在于，所述调用所述人体姿势识别模块的初始化接口并向所述人体姿势识别模块中传入所读取的配置参数信息的步骤包括：

校验所述配置参数；

当所述配置参数校验完成时，初始化所述人体姿势识别模块的配置参数；

根据所述人体姿势识别模块的配置参数的初始化结果，初始化所述人体姿势识别模块中的姿势识别算法模型；

当所述姿势识别算法模型初始化完成时，初始化所述人体姿势识别模块的调试子模块；

当所述调试子模块初始化完成时，调用所述人体关节点坐标检测模块的初始化接口并传入所述配置参数信息；

当所述配置参数信息传入至所述人体关节点坐标检测模块的初始化接口时，调用所述人体关节点坐标检测模块的启动接口。

3.根据权利要求2所述的基于关节点坐标的人体姿势识别方法，其特征在于，所述调用所述人体关节点坐标检测模块的初始化接口并传入所述配置参数信息的步骤包括：

初始化所述人体关节点坐标检测模块的配置参数；

当所述人体关节点坐标检测模块的配置参数初始化完成时，根据丢帧开关初始化预设的丢帧子模块；

根据人体位置校准开关和人体位置校准区域展示开关初始化预设的校准子模块。

4.一种基于关节点坐标的人体姿势识别系统，其特征在于，所述系统包括：

启动模块，用于启动外部应用模块；

外部应用模块，用于读取人体姿势识别模块所需的配置参数信息，调用所述人体姿势识别模块的初始化接口并向所述人体姿势识别模块中传入所读取的配置参数信息；间隔预设时间循环调用所述人体姿势识别模块的获取姿势识别结果接口，获取对应的人体姿势识别结果；

人体姿势识别模块，用于根据所述配置参数信息对待识别图像中的人体姿势进行识别；其中，

所述人体姿势识别模块所需的配置参数包括：Yolo模型类型、Yolo模型文件路径、姿势模型类型、姿势模型文件路径、视频流url、视频流帧缩放后的高、视频流帧缩放后的宽、所使用GPU/CPU的设备号、可视区域比例、丢帧开关、丢帧比例、人体位置校准开关、人体位置校准区域坐标、人体位置校准区域展示开关、录制开关、录制路径、帧保存路径、姿势识别算法模型、姿势识别算法模型文件路径；

所述人体姿势识别模块包括人体关节点坐标检测模块和调试子模块；

所述人体关节点坐标检测模块的配置参数包括：Yolo模型类型、Yolo模型文件路径、姿势模型类型、姿势模型文件路径、视频流url、视频流帧缩放后的高、视频流帧缩放后的宽、所使用GPU/CPU的设备号、丢帧开关、丢帧比例、人体位置校准开关、人体位置校准区域坐标、人体位置校准区域展示开关；其中，

所述人体关节点坐标检测模块包括：

当前帧读取单元，用于通过所述视频流获取和解析服务线程判断当前缓存及所述姿势服务线程状态，若缓存未满且所述姿势服务线程为已完成工作状态，则读取待识别的视频流的当前帧；

丢帧处理单元，用于当所述当前帧读取完成时，判断丢帧开关状态，若丢帧开关状态为开启，则按丢帧比例进行丢帧处理；

处理结果生成单元，用于当所述丢帧处理完成时，根据所述视频流帧缩放后的高和视频流帧缩放后的宽对当前帧进行缩放处理，生成处理结果；

保存及通知单元，用于将所述处理结果保存并通知yolo服务线程；

人像框获取单元，用于所述yolo服务线程调用yolo底层算法接口计算出所述当前帧的人像框；

最优人像框获取单元，用于根据校准框使用算法从所述人像框中筛选出最优人像框并获取处理结果；

最优人像框处理单元，用于将当前人像框的处理结果保存并通知所述姿势服务线程；

人体姿势节点坐标获取单元，用于所述姿势服务线程调用姿势底层算法接口计算出人体姿势节点坐标；

节点坐标反馈单元，用于若所述姿势节点坐标符合预设条件，则通过回调函数将所述姿势节点坐标结果传给所述人体姿势识别模块；

通知单元，用于将上述处理结果保存并通知视频流获取和解析服务线程，并循环执行上述单元。