关键词:
图像去雾
图像去雨
低光照增强
目标检测
指针式仪表
仪表检测
摘要:
现有的指针式仪表检测与读数识别算法大都是基于室内场景所提出的,这些方法在室内环境识别精度高,但是到了室外环境,很难达到原有精度。除此之外,恶劣的天气环境,如雾霾、雨天、低光照等也影响着对指针式仪表的精准定位与读数识别的准确率。因此需要对恶劣天气下捕获的指针式仪表图像进行增强,来提升仪表检测与读数识别的准确性。为了更有效地增强图像,提升图像的恢复质量,并提高指针式仪表检测与读数识别的准确率,本文综述了目前已有的相关图像增强算法与指针式仪表检测与读数识别算法,提出了新的图像增强算法和仪表检测与读数识别算法,并建立了一个端到端的框架,紧密耦合了仪表成分检索和仪表读数识别,提高指针式仪表读取示数的准确性和效率,具体如下:
(1)针对现有的去雾算法难以解决图像中雾霾分布不均匀且细节恢复不足的问题,设计了一个基于编码器-解码器架构的图像去雾网络。该网络利用多尺度结构强大的信息传输与特征提取能力,通过改进的特征注意力模块来增强网络对不同尺度特征的关注度和权重分配,从而提高了模型在去雾任务中的性能表现。此外,通过引入优化模块,对图像在不同尺度上进行重构,有效地增强了图像的细节信息。通过与其他去雾算法的结果进行比较,本文提出的算法在相关数据集上的平均PSNR相比于其他算法的SOTA提升了1.85dB,去雾视觉效果也优于其他算法。
(2)针对目前去雨算法难以捕获雨纹的分布信息的问题,设计了一个多尺度多阶段图像去雨网络。为了更好的将雨纹除去,本文将雨天图像分成不同的块,来提取雨天图像的局部特征,并且设计了双重注意力模块,利用注意力机制有目的地提取各通道之间的有用信息,并应用到各阶段的编解码器子网络和全局初始分辨率子网络中。另外,通过引入跨阶段特征融合,将前一个阶段丰富的语义特征传递到下一个阶段中去。实验结果表明,本文提出的算法在综合雨天数据集和真实雨天数据集中不仅具有更好地去雨效果,并且图像的细节也得到了增强,平均PSNR比其他算法的SOTA提升了0.67dB。
(3)针对现有低光照增强算法存在的色彩失真和对比度不足等问题,提出了跨分辨率低光照增强网络。该网络采用并行的多分辨率卷积流,并引入跨分辨率信息交换机制来捕获丰富的上下文信息和空间信息。使用自适应核特征融合来自动调节感受野,优化网络的表达能力。在将各分辨率流的特征输入到自适应核特征融合模块之前,对各个流的特征进行聚合,进一步提高网络的特征表达能力。另外,设计了自校准注意力来定位图像中的暗区域,从而解决光照不均匀和部分区域过暗的问题。通过与其他低光照算法进行对比,证明了本文提出的低光照增强算法的有效性,平均PSNR比其他低光照算法的SOTA提升了1.38dB,NIQE下降了0.24。
(4)针对现有的指针式仪表检测与读数识别算法在油田现场环境下准确率低、检测算法冗余、难以检测小目标等问题,对相应的仪表检测与读数识别算法做出改进。通过使用轻量化网络GhostNetV2替换目标检测算法Yolov5的原有主干网络,来实现网络的轻量化。采用较小的检测头来提升网络检测小目标的能力,并通过改变损失函数,提高模型在难以分类的样本上的性能。实验表明,本文提出的指针式仪表检测算法能够精准的检测出现场捕获的仪表图像,mAP值达到了98%,比Yolov5-L提升了5%,并且参数量大大降低,只有5.21M。此外,引入空间转换模块,将检测到的指针式仪表图像转换为更符合人们观感的正视图像,并设立一个端到端架构,紧密耦合仪表成分检索和仪表读数识别,从而提高仪表读数的准确性。