基于对数变换和多尺度特征融合的螺栓盒处图像增强

Image Enhancement of Bolt Box Areas Based on Logarithmic Transformation and Multi-scale Feature Fusion

摘要

针对建筑机器人视觉采集装配式建筑单元的螺栓连接盒图像时存在的整体昏暗、螺栓杆处明暗不均和纹理不清晰等问题,提出了一种基于对数变换和多尺度特征融合的图像增强算法。针对螺栓连接盒整体昏暗环境,采用对数变换方法,结合Gamma校正对图像动态扩展优化。通过动态调整参数,提升图像在不同光照条件下的表现,恢复潜在的细节信息。引入多尺度特征融合方法,将图像在不同尺度下的信息通过动态权重分配进行融合处理,从而显著提升图像的纹理清晰度。针对螺栓杆区域,采用对比度受限自适应直方图均衡化(CLAHE)进行强化处理。对图像进行边缘锐化和噪声抑制,确保增强的图像平滑且细节清晰。与当前主流图像增强算法进行对比,本文算法在亮度、清晰度和细节纹理上表现更优,结构相似性、峰值信噪比平均值分别提高了5.89%和4.73%,均方误差降低了12.55%。

Abstract

To address the issues encountered in robotic visual acquisition of bolt connection box images in prefabricated building units (such as overall low brightness, uneven illumination on bolt rods, and unclear texture details), this paper proposes an image enhancement algorithm based on logarithmic transformation and multi-scale feature fusion. For the problem of overall dim lighting in bolt connection box images, a logarithmic transformation is adopted and combined with Gamma correction to optimize dynamic range expansion. By adaptively adjusting parameters, the proposed method improves image performance under varying illumination conditions and effectively recovers latent detail information. Furthermore, a multi-scale feature fusion strategy is introduced, in which image information at different scales is integrated through dynamic weight allocation, thereby significantly enhancing texture clarity. Targeted enhancement is applied to the bolt rod region using the Contrast Limited Adaptive Histogram Equalization (CLAHE) to improve local contrast. In addition, edge sharpening and noise suppression techniques are employed to ensure that the enhanced images remain smooth while preserving fine details. Comparative experiments with mainstream image enhancement algorithms demonstrate that the proposed method achieves superior performance in terms of brightness, clarity, and texture detail. Specifically, the average Structural Similarity Index Measure (SSIM) and Peak Signal-to-Noise Ratio (PSNR) are improved by 5.89% and 4.73%, respectively, while the Mean Squared Error (MSE) is reduced by 12.55%.

关键词

螺栓图像增强 / 对数变换 / 多尺度特征融合 / Gamma校正 / CLAHE增强

Key words

bolt image enhancement / logarithmic transformation / multi-scale feature fusion / Gamma correction / CLAHE enhancement

参考文献

[1] Mustafa W A, Abdul Kader M M M . A review of histogram equalization techniques in image enhancement application [J]. Journal of Physics: Conference Series, 2018, 1019: 012026.

[2] 江巨浪, 张佑生, 薛峰, 等 . 保持图像亮度的局部直方图均衡算法[J]. 电子学报, 2006, 34(5): 861-866.

[3] Jiang Julang, Zhang Yousheng, Xue Feng, et al. Local histogram equalization with brightness preservation [J]. Acta Electronica Sinica, 2006, 34(5): 861-866.

[4] Tan S F, Isa N A M . Exposure based multi-histogram equalization contrast enhancement for non-uniform illumination images[J]. IEEE Access, 2019, 7: 70842-70861.

[5] 张雅荔, 丁少博, 李昌禄, 等 . 基于同态频分聚合的低照度图像增强算法[J]. 激光与光电子学进展, 2023, 60(14): 1410009.

[6] Zhang Yali, Ding Shaobo, Li Changlu, et al. Low-light image-enhancement algorithm based on homomorphic frequency division aggregation [J]. Laser & Optoelectronics Progress, 2023, 60(14): 1410009.

[7] 马红强, 马时平, 许悦雷, 等 . 基于深度卷积神经网络的低照度图像增强[J]. 光学学报, 2019, 39(2): 0210004.

[8] Ma Hongqiang, Ma Shiping, Xu Yuelei, et al. Low-light image enhancement based on deep convolutional neural network [J]. Acta Optica Sinica, 2019, 39(2): 0210004.

[9] Liu Honghai, Hou Xianghua . An illumination compensation algorithm based on single-scale retinex and its application in face recognition [J]. Advanced Materials Research, 2012, 546/547: 755-759.

[10] Li Jiafeng, Hao Shuai, Li Tianshuo, et al. RDMA: low-light image enhancement based on retinex decomposition and multi-scale adjustment [J]. International Journal of Machine Learning and Cybernetics, 2024, 15(5): 1693-1709.

[11] Fu Xueyang, Zeng Delu, Huang Yue, et al. A weighted variational model for simultaneous reflectance and illumination estimation [C]// 2016 IEEE Conference on Computer Vision and Pattern Recognition (CVPR), 2016: 2782-2790.

[12] Lore K G, Akintayo A, Sarkar S . LLNet: a deep autoencoder approach to natural low-light image enhancement [J]. Pattern Recognition, 2017, 61: 650-662.

[13] Wei Chen, Wang Wenjing, Yang Wenhan, et al. Deep retinex decomposition for low-light enhancement [PP/OL]. V1. arXiv (2018-08-14) [2024-10-04]. https://doi.org/10.48550/arXiv.1808.04560.

[14] Yue Guanghui, Gao Jie, Cong Runmin, et al. Deep pyramid network for low-light endoscopic image enhancement [J]. IEEE Transactions on Circuits and Systems for Video Technology, 2024, 34(5): 3834-3845.

[15] Guo Chunle, Li Chongyi, Guo Jichang, et al. Zero-reference deep curve estimation for low-light image enhancement [C]// 2020 IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR), 2020: 1777-1786.

[16] Gupta S, Singh A, Sharma A . Dynamic range adjustment of HDR images using adaptive method [C]// 2022 IEEE 9th Uttar Pradesh Section International Conference on Electrical, Electronics and Computer Engineering (UPCON), 2022: 1-5.

[17] 丁毅, 李玉惠, 李勃 . 基于图像不同亮度区域特征的Gamma矫正方法[J]. 计算机技术与发展, 2016, 26(6): 37-39+45.

[18] Ding Yi, Li Yuhui, Li Bo . Gamma correction based on different brightness regional features for images [J]. Computer Technology and Development, 2016, 26(6): 37-39+45.

[19] Fu Qingqing, Celenk M, Wu Aiping . An improved algorithm based on CLAHE for ultrasonic well logging image enhancement [J]. Cluster Computing, 2019, 22(S5): 12609-12618.

[20] Qiu Xiaohua, Li Min, Zhang Liqiong, et al. Guided filter-based multi-focus image fusion through focus region detection [J]. Signal Processing: Image Communication, 2019, 72: 35-46.

[21] Liu Jun, Yan Ming, Zeng Jinshan, et al. Image smoothing via gradient sparsity and surface area minimization [C]// 2019 IEEE International Conference on Image Processing (ICIP), 2019: 1114-1118.

[22] Sara U, Akter M, Uddin M S . Image quality assessment through FSIM, SSIM, MSE and PSNR-a comparative study [J]. Journal of Computer and Communications, 2019, 7(3): 8-18.

[23] Guo Xiaojie, Li Yu, Ling Haibin . LIME: low-light image enhancement via illumination map estimation [J]. IEEE Transactions on Image Processing, 2017, 26(2): 982-993.

[24] Kwon D, Kim G, Kwon J . DALE: dark region-aware low-light image enhancement [C]// Proceedings of the British Machine Vision Conference 2020, 2020: 196.

基金资助

天津市科技计划项目(22YDTPJC00970)

2023年广东省教育厅普通高校重点领域专项项目(2023ZDZX3107)