本发明涉及人工智能领域,尤其是涉及一种基于计算机视觉的高完整性工人施工姿态检测方法及系统。
背景技术:
1、随着人工智能快速发展,gpu等硬件不断迭代,算力逐渐增强,人工智能技术在工程中地各个领域应用都日趋广泛。在土木工程施工中也有大量人工智能算法应用,例如卷积神经网络、循环神经网络等等。
2、工人的施工动作、施工状态对于土木工程施工具有显著意义,通过识别工人施工状态可以判断工人是否具有安全风险、操作是否规范,也可以更有效地了解施工进度。
3、现有的施工组织管理技术中,通常由施工现场管理人员进行监控,但由于管理人员数量和精力有限,难免出现纰漏。以往的行为检测技术如yolo等,只能识别工人本身,而无法识别其具体动作和施工状态。
4、现有的工人施工姿态,单纯基于openpose算法会导致姿态识别稳定性差,骨架节点完整性差等问题。
技术实现思路
1、本发明的目的就是为了提供一种提高姿态识别准确率的基于计算机视觉的高完整性工人施工姿态检测方法及系统。
2、本发明的目的可以通过以下技术方案来实现:
3、一种基于计算机视觉的高完整性工人施工姿态检测方法,包括以下步骤:
4、s1、实时获取待测施工人员图像数据;
5、s2、将所述待测施工人员图像数据输入至预先构建好的骨架分类网络模型中,提取工人骨架信息;
6、s3、判断所述工人骨架信息是否完整,若是,则执行步骤s5,进行姿态分类,若否,则执行步骤s4,进行工人骨架信息补全操作;
7、s4、将不完整的工人骨架信息输入至预先训练好的骨架补全网络模型中,输出完整的工人骨架信息;
8、s5、将完整的工人骨架信息输入至预先构建好的姿态识别网络模型中,输出工人施工姿态。
9、进一步地,所述骨架分类网络模型的构建过程具体包括:
10、获取工人不同施工姿态图像数据集;
11、对所述图像数据集中的工人的各个关节点进行标注;
12、将标注好的图像数据集输入至cnn网络模型中进行训练,以构建骨架分类网络模型。
13、进一步地,所述标注好的图像数据集在输入至cnn网络模型前,进行标准化和增强处理。
14、进一步地,采用openpose标注各个关节点。
15、进一步地,所述关节点包括头、颈、右肩、右肘、右手、左肩、左肘、左手、右腿、右膝、右脚、左腿、左膝、左脚、右眼、左眼、右耳和左耳。
16、进一步地,采用cgan网络模型构建所述骨架补全网络模型。
17、进一步地,所述姿态识别网络模型采用cnn网络模型进行构建。
18、进一步地,所述工人施工姿态包括站、坐、跑、弯腰、下蹲和抬手动作。
19、进一步地,还包括将所述工人施工姿态进行可视化展示。
20、本发明还提供一种基于计算机视觉的高完整性工人施工姿态检测系统,包括:
21、图像获取模块:用于实时获取待测施工人员图像数据;
22、骨架信息提取模块:用于将所述施工人员图像数据输入至预先构建好的骨架分类网络模型中,提取工人骨架信息;
23、判断模块:用于判断所述工人骨架信息是否完整;
24、补全模块:用于将不完整的工人骨架信息输入至预先训练好的骨架补全网络模型中,输出完整的工人骨架信息;
25、输出模块:用于将完整的工人骨架信息输入至预先构建好的姿态识别网络模型中,输出工人施工姿态。
26、与现有技术相比,本发明具有以下有益效果:
27、(1)相比于传统的单纯采用openpose框架识别骨架节点,本发明引入了骨架补全网络模型对不完整的骨架信息进行补全,生成与原骨架类似的完整骨架,从而可以输出更完整的骨架识别结果,提高施工姿态识别的准确率。
28、(2)本发明能够获得施工现场完整的工人骨架信息,为后续的姿态识别过程提供了更高的稳定性。
29、(3)本发明基于人工智能理论准确地、高完整性地监测施工现场工人的施工状态,减少了施工时对于监管人员如安全员等的人力需求。
1.一种基于计算机视觉的高完整性工人施工姿态检测方法,其特征在于,包括以下步骤:
2.根据权利要求1所述的一种基于计算机视觉的高完整性工人施工姿态检测方法,其特征在于,所述骨架分类网络模型的构建过程具体包括:
3.根据权利要求2所述的一种基于计算机视觉的高完整性工人施工姿态检测方法,其特征在于,所述标注好的图像数据集在输入至cnn网络模型前,进行标准化和增强处理。
4.根据权利要求2所述的一种基于计算机视觉的高完整性工人施工姿态检测方法,其特征在于,采用openpose标注各个关节点。
5.根据权利要求2所述的一种基于计算机视觉的高完整性工人施工姿态检测方法,其特征在于,所述关节点包括头、颈、右肩、右肘、右手、左肩、左肘、左手、右腿、右膝、右脚、左腿、左膝、左脚、右眼、左眼、右耳和左耳。
6.根据权利要求1所述的一种基于计算机视觉的高完整性工人施工姿态检测方法,其特征在于,采用cgan网络模型构建所述骨架补全网络模型。
7.根据权利要求1所述的一种基于计算机视觉的高完整性工人施工姿态检测方法,其特征在于,所述姿态识别网络模型采用cnn网络模型进行构建。
8.根据权利要求1所述的一种基于计算机视觉的高完整性工人施工姿态检测方法,其特征在于,所述工人施工姿态包括站、坐、跑、弯腰、下蹲和抬手动作。
9.根据权利要求1所述的一种基于计算机视觉的高完整性工人施工姿态检测方法,其特征在于,还包括将所述工人施工姿态进行可视化展示。
10.一种基于计算机视觉的高完整性工人施工姿态检测系统,其特征在于,包括: