Abstract
In view of the frequent occurrence of waterlogging in urban areas and the problems that traditional waterlogging monitoring methods consume a lot of human and material resources with high cost and low timeliness, an improved YOLOv5 waterlogging detection method for road traffic is proposed, which enhances the feature extraction of road traffic waterlogging information by feature extraction of waterlogging in urban waterlogging scenarios, and adds the CBAM attention mechanism in the backbone network; and adds a CIoU loss function to optimize the model in the prediction layer to improve the identification accuracy of road traffic waterlogging so as to construct a road traffic waterlogging detection model. In the prediction layer, a CIoU loss function is added to optimize the model and improve the detection accuracy of road water, thus constructing a road water detection model. By screening 5000 road traffic waterlogging images on the public dataset RSCD for training, the experimental results show that the average accuracy of the method is 84.4%, which is 3.7% higher than the original YOLOv5 algorithm, and it can more accurately extract and identify the waterlogged area of the image automatically, which can pave the way for further development of related research, and provide technical support for urban waterlogging monitoring and emergency management. The method can pave the way for further related research and provide technical support for urban flood monitoring and emergency management.
This is a preview of subscription content, log in via an institution to check access.
Access this chapter
Tax calculation will be finalised at checkout
Purchases are for personal use only
References
Lin, T., Liu, X., Song, J., et al.: Urban waterlogging risk assessment based on internet open data: a case study in China. Habitat Int. 71, 88–96 (2018)
Zhang, X., Hu, M., Chen, G., et al.: Urban rainwater utilization and its role in mitigating urban waterlogging problems—a case study in Nanjing, China. Water Resour. Manage 26, 3757–3766 (2012)
Jiang, J., Liu, J., Cheng, C., et al.: Automatic estimation of urban waterlogging depths from video images based on ubiquitous reference objects. Remote Sens. 11(5), 587 (2019)
Liu, Y., Du, M., Jing, C., Cai, G.: Design and implementation of monitoring and early warning system for urban roads waterlogging. In: Li, D., Chen, Y. (eds.) Computer and Computing Technologies in Agriculture VIII. CCTA 2014. IFIP Advances in Information and Communication Technology, vol. 452, pp. 610–615. Springer, Cham (2015). https://doi.org/10.1007/978-3-319-19620-6_68
Redmon, J., Divvala, S., Girshick, R., et al.: You only look once: Unified, real-time object detection. In: Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, pp. 779–788 (2016)
Redmon, J., Farhadi, A.: YOLO9000: better, faster, stronger. In: Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, pp. 7263–7271 (2017)
Al-Haija, Q.A., Smadi, M., Al-Bataineh, O.M.: Identifying phasic dopamine releases using DarkNet-19 convolutional neural network. In: 2021 IEEE International IoT, Electronics and Mechatronics Conference (IEMTRONICS), pp. 1–5. IEEE (2021)
Redmon, J., Farhadi, A.: YOLOv3: an incremental improvement. arXiv:1804.02767 (2018)
Lin, T.Y., Dollár, P., Girshick, R., et al.: Feature pyramid networks for object detection. In: Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, pp. 2117–2125 (2017)
Bochkovskiy, A., Wang, C.Y., Liao, H.YM.: Yolov4: Optimal speed and accuracy of object detection. arXiv preprint arXiv:2004.10934 (2020)
Wang, C.Y., Liao, H.Y.M., Wu, Y.H., et al. CSPNet: a new backbone that can enhance learning capability of CNN. In: Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition Workshops, pp. 390–391 (2020)
Shi, X., Hu, J., Lei, X., et al.: Detection of flying birds in airport monitoring based on improved YOLOv5. In: 2021 6th International Conference on Intelligent Computing and Signal Processing (ICSP), pp. 1446–1451. IEEE (2021)
Yang, J., Fu, X., Hu, Y., et al.: PanNet: a deep network architecture for pan-sharpening. In: Proceedings of the IEEE International Conference on Computer Vision, pp. 5449–5457 (2017)
Woo, S., Park, J., Lee, J.Y., et al.: Cbam: convolutional block attention module. In: Proceedings of the European Conference on Computer Vision (ECCV), pp. 3–19 (2018)
Rezatofighi, H., Tsoi, N., Gwak, J.Y., et al.: Generalized intersection over union: a metric and a loss for bounding box regression. In: Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition, pp. 658–666 (2019)
Yu, J., Jiang, Y., Wang, Z., et al.: Unitbox: an advanced object detection network. In: Proceedings of the 24th ACM International Conference on Multimedia, pp. 516–520 (2016)
Zheng, Z., Wang, P., Liu, W., et al.: Distance-IoU loss: faster and better learning for bounding box regression. In: Proceedings of the AAAI Conference on Artificial Intelligence, vol. 34, no. 07, pp. 12993–13000 (2020)
Acknowledgments
This work was supported by the Jiangsu Provincial College Students Innovation and Entrepreneurship Training Plan Project (grant number 202311276103Y), National Natural Science Foundation of China (grant number 41972111) and the Second Tibetan Plateau Scientific Expedition and Research Program (STEP) (grant number 2019QZKK020604).
Author information
Authors and Affiliations
Corresponding author
Editor information
Editors and Affiliations
Rights and permissions
Copyright information
© 2024 The Author(s), under exclusive license to Springer Nature Singapore Pte Ltd.
About this paper
Cite this paper
Liu, J., Shang, Y., Li, X., Hao, H., Geng, P. (2024). Road Traffic Waterlogging Detection Based on YOLOv5. In: Jin, H., Pan, Y., Lu, J. (eds) Data Science and Information Security. IAIC 2023. Communications in Computer and Information Science, vol 2059. Springer, Singapore. https://doi.org/10.1007/978-981-97-1280-9_4
Download citation
DOI: https://doi.org/10.1007/978-981-97-1280-9_4
Published:
Publisher Name: Springer, Singapore
Print ISBN: 978-981-97-1279-3
Online ISBN: 978-981-97-1280-9
eBook Packages: Computer ScienceComputer Science (R0)