本发明涉及计算机视觉领域,尤其涉及一种工业设备指示灯状态识别方法。
背景技术:
1、随着人工智能技术和物联网技术的发展,越来越多的设备和系统需要进行监控和管理。其中,设备状态指示灯作为重要信息来源,常使用包括红、蓝、绿等颜色来展示设备运行状态。
2、为了满足即时运行控制的要求,获得更高效的运行控制策略,设备状态的监测识别也不断向智能化方向转变,
3、目前工业上主要是通过安装固定的辅助摄像头或设计巡检机器人采集设备上包括指示灯的仪器仪表图像视频,并在此基础上做进一步识别工作。
4、在设备指示灯识别技术方面,目前也有研究者对yolo系列算法的应用,如使用yolo v3模型确定设备位置后,首先采用预先设定好的图像比例确定设备中指示灯位置,并裁剪出来。再将图像进行色域变化,转换到hsv空间,然后设定红、绿、黄三种颜色的阈值进行二值化,最后选取特定颜色查找指示灯轮廓并计数。或者使用yolov5算法识别指示灯位置,再使用色值筛选和二值化方法识别状态。
5、目前进行设备状态识别普遍缺少操作简便、适应性强、识别精度高的检测方法。其中使用辅助摄像头识别仍是较为主流的方法,如使用openmv摄像头模块自带的颜色识别方法对工作指示灯进行状态识别,但其需事先通过openmv ide软件向openmv摄像头模块写入最新颜色阈值,操作较为繁琐,每台设备都需要根据环境进行手动调整。巡检机器人的使用在室外使用较少,且容易受路线障碍阻碍,机器人本身只能拍摄图片,仍需要搭载相应识别技术或者回收照片进行人工识别,而人工识别效率较低,且难以保证实时性和准确性检测要求。
6、而在设备上所搭载的技术多仅使用yolo模型识别指示灯或设备的位置,再结合其它方法,例如传统的色域变化、二值化方法来确定指示灯的亮灭状态,或是通过其它神经网络进行类别和状态的识别,仍存在技术应用上的一些局限性,需要进一步探索和改进,以提高识别准确度和可靠性。
技术实现思路
1、本发明提供了一种工业设备指示灯状态识别方法,具备高效的、能够快速准确地判断出设备指示灯类别、位置以及设备状态的优点,解决了现有技术中存在的准确度和可靠性差的问题。
2、根据本申请实施例提供的一种工业设备指示灯状态识别方法,包括如下步骤:
3、获取设备指示灯图片数据,对数据进行预处理,清理较低质量的数据,打好标签后并按照一定比例随机划分为训练集、测试集和检验集;
4、将训练数据带入改进的yolov7-tiny模型中,通过迭代调整网络模型参数;
5、yolov7模型通过由多层卷积层与elan-t层构成的主干网络和加强特征提取网络提取特征,通过检测头进行指示灯状态识别;
6、利用测试集图像对训练好的模型测试,模型可以输出工业设备指示灯状态信息。
7、优选地,上述获取设备指示灯图片数据步骤具体为:
8、获取数据集,获取多张设备指示灯图片,包括不同角度、不同光照条件下的设备指示灯图像,并且每个图像应该有相应的标签来表示该图像中出现的指示灯状态;
9、基于此建立初始图像数据集,并进行预处理,去除一些异常、分辨率低的图片,并将图像数据集按照一定比例随机划分为训练集,验证集和测试集。
10、优选地,上述步骤还包括:
11、网络及结构改进,yolov7-tiny网络结构包括输入层,主干网络和头部网络三部分;
12、输入层,使用mosaic数据增强,将四张图片进行随机缩放裁剪等操作,并拼接起来以丰富数据集;
13、主干网络,包括bconv层、mp层以及多层级特征融合模块,结合空间金字塔池化模块;
14、头部网络,yolov7模型将neck部分和head部分进行了合并,以sppcsp为基准,分为五个阶段,前四个阶段包括了卷积层、spp层、csp层和上采样层,用于提取特征并进行特征融,对引入的sppcsp结构,通过空间金字塔池化和卷积操作,实现了对不同尺度特征的提取和融合,最终通过三个卷积层,用于预测每个网格单元格的边界框坐标、物体得分和类别概率,在头部网络中的两个上采样层之后添加cbam注意力机制,由通道注意力和空间注意力两部分组成,可以自适应地融合特征;
15、回归框损失函数,为使模型能够在训练时减少受到低质量样本的影响,引入基于动态的非单调焦点机制wise-iou损失函数;
16、选择训练模型,使用改进的yolov7-tiny算法来实现指示灯状态识别。
17、本申请实施例提供的技术方案可以包括以下有益效果:
18、1、提高检测精度:与已有技术相比,本发明采用yolov7-tiny结合cbam注意力机制进行改进,依靠更为高效、准确的特征提取方法,提高了目标检测的精度和召回率。
19、2、减少计算成本:本发明的模型架构经过优化,采用wise-iou这类结构更为简单,参数量更少(改进模型在参数量减少83.5%)的回归框损失函数,具有更小的模型大小和更快的推理速度,能够在保证检测精度的同时,降低计算成本,适用于边缘设备等资源受限的场景。
20、3、支持多种目标分类检测:本发明的模型支持多类目标的检测和分类,一次性完成目标识别与检测,并且可以根据实际需求进行灵活配置和扩展,具有较强的通用性和可扩展性。
21、4、本发明涉及一种工业设备指示灯状态识别方法,该方法通过引入cbam注意力模块和改进iou损失函数,为基于动态非单调聚焦的wise-iou函数,提高了模型的目标识别能力和边界框损失的拟合效果。针对设备指示灯数据集特点,对yolov7模型的训练要求进行了相应调整。解决了目前工业设备指示灯状态识别中存在的精度低、效率低等问题,具有较强的实用性和广泛的应用前景。此技术方案可以为工业设备状态检测提供有效的支持,为企业降低生产成本,提高生产效率提供了有力的保障。此外,本研究的方法也可以应用到其他领域的图像识别和目标检测任务中,例如自动驾驶、智能安防、医疗影像等领域,具有重要的理论和实践意义。
1.一种工业设备指示灯状态识别方法,其特征在于,包括如下步骤:
2.根据权利要求1所述的一种工业设备指示灯状态识别方法,其特征在于,上述获取设备指示灯图片数据步骤具体为:
3.根据权利要求1所述的一种工业设备指示灯状态识别方法,其特征在于,上述步骤还包括: