摘要
本发明提出了基于图像序列和Transformer网络的工业设备指示灯状态识别方法,所述方法设置目标检测网络和状态识别网络,所述目标检测网络把检测到的所有指示灯子图像组合成图像序列,一次性同时输出所有指示灯状态;所述状态识别网络基于Transformer结构的网络,分别计算每个指示灯子图像内部的特征之间的关系和计算所有指示灯图像之间的特征的关系,综合的识别指示灯的状态;所述方法以图像序列的形式一次性输入到神经网络模型中,通过学习单个指示灯内部的图像特征关系和每个指示灯之间的图像特征关系,识别每个指示灯的状态,除关注指示灯本身,还考虑全局特征和指示灯之间的对比,使其在复杂光照条件具有更准确的识别效果。
技术关键词
图像编码器
工业设备
序列
图像特征关系
神经网络模型
解码器
输出指示灯
图片
置信度阈值
矩阵
尺寸
网格
元素
注意力机制