一种用于输电工程安全建设的图像识别方法和系统
Abstract:
本发明提出了一种用于输电工程安全建设的图像识别方法和系统,该方法包括获取输电场景下的烟火隐患图像数据,并将烟火隐患图像数据按预设比例划分为训练集和测试集;构建图像识别的检测网络YOLOV5;在网络的SPPF结构后加入transformer模块,并在特征融合层内增加跳跃连接;将训练集数据预处理后输入检测网络,完成对烟火隐患图像数据的检测模型训练;然后对测试集数据进行测试,测试之后得到输电工程安全建设的图像识别结果。基于该方法,还提出了一种用于输电工程安全建设的图像识别系统,本发明修改了检测网络结构,并修改了特征融合的连接方式,加强了图像全局和局部的依赖关系,并增强了特征的提取能力,降低背景干扰。
Patent Agency Ranking
0/0