施工作业检测模型训练方法、装置、设备及存储介质
摘要:
本发明实施例公开了一种施工作业检测模型训练方法、装置、设备及存储介质。该方法包括:确定工程车辆施工作业过程中同时包含工程车辆和施工作业人员的,具有工程车辆标准区域和作业人员标准区域的样本标签的施工作业图像,并生成包括至少一张施工作业图像的样本训练集,并输入至预先构建的包括有效特征提取网络层、特征融合网络层和分类回归网络层的网络模型中,根据模型输出的工程车辆预测区域和工程车辆标准区域,以及作业人员预测区域和作业人员标准区域,进行模型训练,得到满足预设的模型训练结束条件,得到目标检测模型进行施工作业检测。本发明实施例提供的技术方案,基于网络模型进行施工作业的自动检测,以提高检测效率及检测准确度。
0/0