Robot-Based Hazard Detection for Wastewater Treatment Plants
Abstract
1. Introduction
- (1)
- A robot-based hazard detection framework is proposed for WWTPs by integrating mobile robots with object detection to identify typical hazards.
- (2)
- A hazard detection dataset is built from WWTP inspection images and public web images, covering sludge flotation, smoke, fire, standing water, and personnel protection status.
- (3)
- Faster R-CNN, YOLOv8, YOLOv11, and YOLOv12 are systematically compared in terms of detection performance, computational complexity, and inference efficiency to support model selection for robot-based inspection in WWTPs.
2. Related Work
2.1. Hazard Detection Methods in WWTPs
2.2. Intelligent Inspection Robots
2.3. Object Detection Methods
3. Materials and Methods
3.1. Inspection Robot
3.2. Inspection Scenarios and Detection Objects
3.2.1. Secondary Clarifier Scenario
3.2.2. Confined Space Scenario
3.3. Hazard Detection Dataset for WWTPs
3.4. Object Detection Model Selection
3.5. Implementation Details
4. Results
4.1. Quantitative Results in the Secondary Clarifier Scenario
4.2. Quantitative Results in the Confined Space Scenario
4.3. Precision–Recall Curve and Confusion Matrix Analysis
4.4. Evaluation of YOLOv12 Under Different Image Sources
4.5. Grouped Stratified Split Evaluation
4.6. Qualitative Results
4.7. Complexity Analysis
5. Discussion
Author Contributions
Funding
Data Availability Statement
Conflicts of Interest
References
- Qu, J.; Wang, H.; Wang, K.; Yu, G.; Ke, B.; Yu, H.Q.; Ren, H.; Zheng, X.; Li, J.; Li, W.W.; et al. Municipal wastewater treatment in China: Development history and future perspectives. Front. Environ. Sci. Eng. 2019, 13, 88. [Google Scholar] [CrossRef] [Scilit]
- Chen, Q.; Fan, J.; Chen, W.; Zhang, A.; Pan, G. A dimensionality-reducible operational optimal control for wastewater treatment process. IEEE Trans. Neural Netw. Learn. Syst. 2022, 34, 5418–5426. [Google Scholar] [CrossRef] [Scilit]
- Chen, A.; Wu, M. Causal mechanisms of 100 accidents in sewage system facilities. Water Pract. Technol. 2025, 20, 1–14. [Google Scholar] [CrossRef] [Scilit]
- Sancho Martínez, J.; Fernández, Y.B.; Leinster, P.; Casado, M.R. Combining unmanned aircraft systems and image processing for wastewater treatment plant asset inspection. Remote Sens. 2020, 12, 1461. [Google Scholar] [CrossRef] [Scilit]
- Kumar, P.; Chang, T.H.; Zhong, Z.G.; Yao, C.K.; Chang, C.Y.; Chen, C.S.; Su, C.Y.; Peng, P.C. Enhanced Activity Detection in Mechanical Robot Dog Using Dynamic Strain-Based FBG Sensors and YOLO-v7. IEEE Sens. J. 2025, 25, 31572–31582. [Google Scholar] [CrossRef] [Scilit]
- Yang, S.; Behzadian, K.; Coleman, C.; Holloway, T.G.; Campos, L.C. Application of AI-based techniques for anomaly management in wastewater treatment plants: A review. J. Environ. Manag. 2025, 392, 126886. [Google Scholar] [CrossRef] [Scilit]
- Chang, P.; Meng, F.C. Fault detection of urban wastewater treatment process based on combination of deep information and transformer network. IEEE Trans. Neural Netw. Learn. Syst. 2022, 35, 8124–8133. [Google Scholar] [CrossRef] [Scilit]
- Peng, S.; Guo, Y.; Wang, J.; Wang, Y.; Zhang, W.; Zhou, X.; Jiang, L.; Lai, B. The coagulation-precipitation turbidity prediction model for precision drug delivery system based on deep learning and machine vision. J. Environ. Chem. Eng. 2024, 12, 112211. [Google Scholar] [CrossRef] [Scilit]
- Borzooei, S.; Scabini, L.; Miranda, G.; Daneshgar, S.; Deblieck, L.; Bruno, O.; De Langhe, P.; De Baets, B.; Nopens, I.; Torfs, E. Evaluation of activated sludge settling characteristics from microscopy images with deep convolutional neural networks and transfer learning. J. Water Process Eng. 2024, 64, 105692. [Google Scholar] [CrossRef] [Scilit]
- Offir, I.; Moni, S.; Jacob, G.; Ido, C.; Ofir, M.; Dror, A. Analyzing the secondary wastewater-treatment process using Faster R-CNN and YOLOv5 object detection algorithms. J. Clean. Prod. 2023, 416, 137913. [Google Scholar] [CrossRef] [Scilit]
- Chang, P.; Wang, K.; Zheng, K.; Meng, F.C. Monitoring of wastewater treatment process based on multi-stage variational autoencoder. Expert Syst. Appl. 2022, 207, 117919. [Google Scholar] [CrossRef] [Scilit]
- Ha, B.; Schalter, B.; White, L.; Köhler, J. Automatic defect detection in sewer network using deep learning based object detector. arXiv 2024, arXiv:2404.06219. [Google Scholar]
- Jia, Z.; Liu, H.; Zheng, H.; Fan, S.; Liu, Z. An intelligent inspection robot for underground cable trenches based on adaptive 2d-slam. Machines 2022, 10, 1011. [Google Scholar] [CrossRef] [Scilit]
- Fischer, G.K.; Bergau, M.; Gómez-Rosal, D.A.; Wachaja, A.; Graeter, J.; Odenweller, M.; Piechottka, U.; Höflinger, F.; Gosala, N.; Wetzel, N.; et al. Evaluation of a smart mobile robotic system for industrial plant inspection and supervision. IEEE Sens. J. 2024, 24, 19684–19697. [Google Scholar] [CrossRef] [Scilit]
- Gómez-Rosal, D.A.; Bergau, M.; Fischer, G.K.; Wachaja, A.; Grater, J.; Odenweller, M.; Piechottka, U.; Hoeflinger, F.; Gosala, N.; Wetzel, N.; et al. A smart robotic system for industrial plant supervision. In Proceedings of the 2023 IEEE SENSORS; IEEE: Piscataway, NJ, USA, 2023; pp. 1–4. [Google Scholar]
- Shen, W.; Zhang, J.; Jin, W.; Song, A. An RGBD fused real time defect detection and size estimation method for sewer robots. Measurement 2025, 259, 119694. [Google Scholar] [CrossRef] [Scilit]
- Girshick, R.; Donahue, J.; Darrell, T.; Malik, J. Rich feature hierarchies for accurate object detection and semantic segmentation. In Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, Columbus, OH, USA, 23–28 June 2014; pp. 580–587. [Google Scholar]
- Girshick, R. Fast r-cnn. In Proceedings of the IEEE International Conference on Computer Vision, Santiago, Chile, 7–16 December 2015; pp. 1440–1448. [Google Scholar]
- Ren, S.; He, K.; Girshick, R.; Sun, J. Faster r-cnn: Towards real-time object detection with region proposal networks. Adv. Neural Inf. Process. Syst. 2015, 28, 91–99. [Google Scholar]
- Lin, T.Y.; Goyal, P.; Girshick, R.; He, K.; Dollár, P. Focal loss for dense object detection. In Proceedings of the IEEE International Conference on Computer Vision, Venice, Italy, 22–29 October 2017; pp. 2980–2988. [Google Scholar]
- Redmon, J.; Divvala, S.; Girshick, R.; Farhadi, A. You only look once: Unified, real-time object detection. In Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, Las Vegas, NV, USA, 27–30 June 2016; pp. 779–788. [Google Scholar]
- Tian, Y.; Ye, Q.; Doermann, D. Yolov12: Attention-centric real-time object detectors. Adv. Neural Inf. Process. Syst. 2026, 38, 78433–78457. [Google Scholar]








| Subset | Category (Class ID) | Class Images | Annotated Instances | Common Resolutions | Common Aspect Ratios | Total Images | Train/Test |
|---|---|---|---|---|---|---|---|
| Secondary Clarifier | Sludge Flotation (0) | 806 | 1386 | (33%) (9.81%) | 16:9 (38.88%) 4:3 (12.25%) | 1600 | 1280/320 |
| Confined Space | No Mask (0) Mask (1) Fire (2) Smoke (3) Standing Water (4) | 347 845 887 929 1000 | 610 2327 1363 1207 2078 | (17.6%) (9.86%) | 1:1 (30.46%) 4:3 (5.26%) | 3500 | 2797/703 |
| Category | Metrics | Faster R-CNN | YOLOv8 | YOLOv11 | YOLOv12 |
|---|---|---|---|---|---|
| Sludge Flotation | Precision | 0.784 | 0.869 | 0.900 | 0.804 |
| Recall | 0.935 | 0.808 | 0.861 | 0.910 | |
| F1-score | 0.853 | 0.837 | 0.880 | 0.854 | |
| mAP@0.5 | 0.906 | 0.913 | 0.914 | 0.917 | |
| mAP@0.5:0.95 | 0.754 | 0.785 | 0.788 | 0.802 |
| Category | Metrics | Faster R-CNN | YOLOv8 | YOLOv11 | YOLOv12 |
|---|---|---|---|---|---|
| No Mask | Precision | 0.693 | 0.809 | 0.794 | 0.841 |
| Recall | 0.754 | 0.698 | 0.746 | 0.722 | |
| F1-score | 0.722 | 0.749 | 0.769 | 0.777 | |
| mAP@0.5 | 0.688 | 0.741 | 0.778 | 0.783 | |
| mAP@0.5:0.95 | 0.271 | 0.265 | 0.283 | 0.371 | |
| Mask | Precision | 0.857 | 0.928 | 0.919 | 0.929 |
| Recall | 0.875 | 0.851 | 0.876 | 0.854 | |
| F1-score | 0.866 | 0.888 | 0.897 | 0.890 | |
| mAP@0.5 | 0.859 | 0.928 | 0.927 | 0.940 | |
| mAP@0.5:0.95 | 0.378 | 0.428 | 0.414 | 0.532 | |
| Fire | Precision | 0.609 | 0.881 | 0.888 | 0.877 |
| Recall | 0.808 | 0.775 | 0.808 | 0.778 | |
| F1-score | 0.695 | 0.825 | 0.846 | 0.825 | |
| mAP@0.5 | 0.757 | 0.862 | 0.868 | 0.843 | |
| mAP@0.5:0.95 | 0.385 | 0.579 | 0.581 | 0.552 | |
| Smoke | Precision | 0.482 | 0.895 | 0.872 | 0.873 |
| Recall | 0.708 | 0.633 | 0.642 | 0.661 | |
| F1-score | 0.574 | 0.742 | 0.740 | 0.752 | |
| mAP@0.5 | 0.613 | 0.784 | 0.750 | 0.787 | |
| mAP@0.5:0.95 | 0.293 | 0.574 | 0.571 | 0.585 | |
| Standing Water | Precision | 0.347 | 0.871 | 0.828 | 0.860 |
| Recall | 0.366 | 0.617 | 0.662 | 0.620 | |
| F1-score | 0.356 | 0.722 | 0.736 | 0.721 | |
| mAP@0.5 | 0.335 | 0.756 | 0.751 | 0.742 | |
| mAP@0.5:0.95 | 0.162 | 0.431 | 0.434 | 0.465 | |
| All | Precision | 0.603 | 0.877 | 0.860 | 0.876 |
| Recall | 0.697 | 0.715 | 0.747 | 0.727 | |
| F1-score | 0.647 | 0.788 | 0.800 | 0.795 | |
| mAP@0.5 | 0.650 | 0.814 | 0.815 | 0.819 | |
| mAP@0.5:0.95 | 0.298 | 0.455 | 0.457 | 0.501 |
| Dataset | Source | Images | Precision | Recall | F1-Score | mAP@0.5 | mAP@0.5:0.95 |
|---|---|---|---|---|---|---|---|
| Secondary Clarifier | Robot | 141 | 0.817 | 0.930 | 0.870 | 0.919 | 0.781 |
| Fixed-camera | 57 | 0.802 | 0.919 | 0.857 | 0.902 | 0.820 | |
| Public Web | 122 | 0.880 | 0.854 | 0.867 | 0.933 | 0.826 | |
| Confined Space | Robot | 200 | 0.880 | 0.780 | 0.827 | 0.837 | 0.531 |
| Fixed-camera | 239 | 0.851 | 0.762 | 0.804 | 0.844 | 0.514 | |
| Public Web | 264 | 0.891 | 0.685 | 0.775 | 0.782 | 0.474 |
| Dataset | Metrics | Faster R-CNN | YOLOv8 | YOLOv11 | YOLOv12 |
|---|---|---|---|---|---|
| Secondary Clarifier | Precision | 0.762 | 0.799 | 0.891 | 0.799 |
| Recall | 0.902 | 0.889 | 0.873 | 0.856 | |
| F1-score | 0.826 | 0.842 | 0.881 | 0.827 | |
| mAP@0.5 | 0.880 | 0.877 | 0.865 | 0.883 | |
| mAP@0.5:0.95 | 0.689 | 0.759 | 0.749 | 0.776 | |
| Confined Space | Precision | 0.589 | 0.855 | 0.854 | 0.828 |
| Recall | 0.696 | 0.683 | 0.739 | 0.709 | |
| F1-score | 0.638 | 0.760 | 0.792 | 0.764 | |
| mAP@0.5 | 0.666 | 0.784 | 0.723 | 0.801 | |
| mAP@0.5:0.95 | 0.308 | 0.450 | 0.442 | 0.457 |
| Subset | Model | Params (M) | GFLOPs | RTX 2080 Ti Inference Time (ms) | Ascend 910B2 Inference Time (ms) |
|---|---|---|---|---|---|
| Secondary Clarifier | Faster R-CNN | 41.30 | 133.92 | 42.06 | – |
| YOLOv8 | 3.01 | 8.19 | 10.61 | – | |
| YOLOv11 | 2.59 | 6.45 | 13.38 | – | |
| YOLOv12 | 2.55 | 6.44 | 12.99 | 9.21 | |
| Confined Space | Faster R-CNN | 41.32 | 133.94 | 42.65 | – |
| YOLOv8 | 3.01 | 8.20 | 17.12 | – | |
| YOLOv11 | 2.59 | 6.45 | 18.63 | – | |
| YOLOv12 | 2.55 | 6.44 | 17.56 | 12.45 |
Disclaimer/Publisher’s Note: The statements, opinions and data contained in all publications are solely those of the individual author(s) and contributor(s) and not of MDPI and/or the editor(s). MDPI and/or the editor(s) disclaim responsibility for any injury to people or property resulting from any ideas, methods, instructions or products referred to in the content. |
© 2026 by the authors. Licensee MDPI, Basel, Switzerland. This article is an open access article distributed under the terms and conditions of the Creative Commons Attribution (CC BY) license.
Share and Cite
Liu, H.; Ji, Z.; Li, B.; Feng, H.; Zhang, W.; Zhang, Z.; Kong, W.; Ni, G. Robot-Based Hazard Detection for Wastewater Treatment Plants. Electronics 2026, 15, 3801. https://doi.org/10.3390/electronics15173801
Liu H, Ji Z, Li B, Feng H, Zhang W, Zhang Z, Kong W, Ni G. Robot-Based Hazard Detection for Wastewater Treatment Plants. Electronics. 2026; 15(17):3801. https://doi.org/10.3390/electronics15173801
Chicago/Turabian StyleLiu, Hui, Zhenyan Ji, Bin Li, Haojie Feng, Wenqi Zhang, Zhipeng Zhang, Weiheng Kong, and Guohao Ni. 2026. "Robot-Based Hazard Detection for Wastewater Treatment Plants" Electronics 15, no. 17: 3801. https://doi.org/10.3390/electronics15173801
APA StyleLiu, H., Ji, Z., Li, B., Feng, H., Zhang, W., Zhang, Z., Kong, W., & Ni, G. (2026). Robot-Based Hazard Detection for Wastewater Treatment Plants. Electronics, 15(17), 3801. https://doi.org/10.3390/electronics15173801

