Welcome to Francis Academic Press

Academic Journal of Computing & Information Science, 2023, 6(12); doi: 10.25236/AJCIS.2023.061204.

An Improved Method for Digital Water Meter Reading Area Segmentation Based on U~2-Net

Author(s)

Rongrong Fei, Runze Cheng, Bin Yao, Feng Tian, Liuyang Gao, Zemu Men

Corresponding Author:
Rongrong Fei
Affiliation(s)

School of Electronic Information and Artificial Intelligence, Shaanxi University of Science and Technology, Xi’an, China

Abstract

With the continuous development of intelligent technology, the recognition of water meter readings has brought convenience to people's lives. By automatically reading water meter data, work efficiency can be improved. However, due to the varying shapes and sizes of the recognition areas, it is difficult to achieve good recognition results. Therefore, in response to the problem of varying sizes of water meter reading recognition areas, this paper proposes a method for digital water meter reading recognition area segmentation based on an improved U2-Net. Firstly, in terms of capturing data features, an improved Double-RSU module based on the RSU module is designed. This module increases depth and complexity compared to the original RSU module, thereby improving generalization and robustness. Secondly, in terms of model training, a combination of cross-entropy loss function, Jaccard coefficient and Dice loss function is used to comprehensively evaluate the entire model by considering binary classification, segmentation accuracy and segmentation area overlap. Experimental results show that the proposed method has good performance in MIoU, MAE and F1Score indicators, with the MIoU indicator increasing from 0.878 to 0.91, the MAE indicator increasing from 0.001 to 0.0009, and the F1Score indicator increasing from 0.9431 to 0.9455. These results indicate that the improved model has better performance compared to the original U2-Net model and can more accurately perform water meter reading area segmentation.

Keywords

Salient Object Detection, Deep Learning, Image Segmentation, U2-Net

Cite This Paper

Rongrong Fei, Runze Cheng, Bin Yao, Feng Tian, Liuyang Gao, Zemu Men. An Improved Method for Digital Water Meter Reading Area Segmentation Based on U~2-Net. Academic Journal of Computing & Information Science (2023), Vol. 6, Issue 12: 33-44. https://doi.org/10.25236/AJCIS.2023.061204.

References

[1] Shen J F. Research and application of water meter reading recognition method based on deep learning [D]. Jishou University, 2022.

[2] Xiong L. Water meter reading recognition method based on deep learning in real scene[D]. Nanchang University, 2022.

[3] Cui Y. Reading recognition method of digital water meter image based on deep learning[D]. Nanchang University, 2022.

[4] Li Y Z, Zhao J S. A review of image saliency object detection based on deep learning[J]. Software Engineering, 2023, 26(01): 1-4.

[5] Zhang L L. Research and implementation of water meter reading recognition algorithm[D]. Chongqing University of Posts and Telecommunications, 2022.

[6] Ronneberger O, Fischer P, Brox T. U-Net: Convolutional Networks for Biomedical Image Segmentation [J]. International Conference on Medical Image Computing and Computer-assisted Intervention, 2015, 9351: 234-241.

[7] Zhou Z W, Siddiquee M M R, Tajbakhsh N, et al. UNet++: A Nested U-Net Architecture for Medical Image Segmentation[C]. Deep Learning in Medical Image Analysis and Multimodal Learning for Clinical Decision Support, 2018, 11045: 3-11.

[8] Qin X, Zhang Z, Huang C, et al. U2-Net: Going deeper with nested U-structure for salient object detection [J]. Pattern Recognition, 2020, 106.

[9] Wang Y X, Ge H W. Metal surface defect detection algorithm based on U2-Net[J]. Journal of Nanjing University (Natural Sciences), 2023, 59(03): 413-424.

[10] Ye M, Li X C, Liu K, et al. An impedance imaging method based on the U2-Net model[J]. Chinese Journal of Scientific Instrument, 2021, 42(02): 235-243.

[11] Jha D, Riegler M A, Johansen D, et al. DoubleU-Net: A Deep Convolutional Neural Network for Medical Image Segmentation[J]. IEEE 33rd International Symposium on Computer-Based Medical Systems, 2020: 558-564.

[12] Simonyan K, Zisserman A. Very Deep Convolutional Networks for Large-Scale Image Recognition [J]. Computer Science, 2014.

[13] Hou X X. Research on highland lake extraction of medium-resolution remote sensing images based on DoubleU-Net[D]. Guangzhou University, 2022.

[14] Chen L C, Papandreou G, Schroff F, et al. Rethinking Atrous Convolution for Semantic Image Segmentation [J]. Computer Science, 2017.

[15] Zhao H S, Shi J P, Qi X J, et al. Pyramid Scene Parsing Network[J]. Computer Science, 2017: 6230-6239.

[16] Abhishek C, Eugenio C. LinkNet: Exploiting Encoder Representations for Efficient Semantic Segmentation [J]. IEEE Visual Communications and Image Processing, 2017.

[17] Chen L C, Zhu Y k, Papandreou G, et al. Encoder-Decoder with Atrous Separable Convolution for Semantic Image Segmentation[J]. European Conference on Computer Vision, 2018, 11211: 833-851.

[18] Liang J Y, Sun G L, Zhang K, et al. Mutual Affine Network for Spatially Variant Kernel Estimation in Blind Image Super-Resolution[J]. IEEE/CVF International Conference on Computer Vision, 2021: 4076-4085.

[19] Li H C, Xiong P F, An J, et al. Pyramid Attention Network for Semantic Segmentation[J]. Computer Science, 2018.