在审

电力生产场景下的人员行为检测模型训练方法、装置、设备、介质及产品

喻召杰、王强、刘兆平、王勇军、钟宇欢、陈伟、李锐
南网数字运营软件科技(广东)有限公司

摘要

本发明公开了一种电力生产场景下的人员行为检测模型训练方法、装置、设备、介质及产品。该方法包括:基于电力生产场景下的历史人员行为图像构建训练样本图像;将训练样本图像输入至预先构建的人员行为检测模型中的骨干网络进行特征提取,得到特征提取图像;将特征提取图像输入至颈部网络进行特征融合处理,得到特征融合图像;将特征融合图像输入至头部网络进行目标位置与类别预测,得到预测人员位置、预测安全帽位置和预测手持工具位置及预测手持工具类别;根据训练样本图像的预测人员位置、预测安全帽位置和预测手持工具位置及预测手持工具类别对人员行为检测模型进行模型训练,直到满足模型训练结束条件,得到已完成训练的人员行为检测模型。

暂无引用专利