一种基于YOLOv4改进算法的电力智慧工地违规行为检测方法

摘要

本发明公开了一种基于YOLOv4改进算法的电力智慧工地违规行为检测方法。采集用于训练模型的电力施工现场的图像；对采集到的电力施工现场的图像进行图像增强；将图像及数据增强后的得到的目标区域用矩形框进行标注，获取矩形框的坐标与所包含的种类；根据采集到的施工现场的图像及数据增强后得到的图像，及获取的所述矩形框的坐标与包含的种类对基于YOLOv4改进的模型进行训练；实时采集电力施工现场的图像，根据训练好的模型和实时采集到待检测的电力施工现场图像，并输出违规图像。本方法适用于施工现场的人员违规行为与施工车辆违规操作的检测，实现电力智慧工地可视、精细、智能管理，有效提升工程现场管理水平并降低安全风险。