Abstract:A method based on YOLOv8 for water instance segmentation has been proposed, achieving rapid, efficient, and accurate measurement of the dry beach length of tailings ponds under real-time video streams. Firstly, a high-quality water instance segmentation COCO dataset is completed. Secondly, mainstream deep learning instance segmentation algorithms are analyzed, and the YOLOv8 model is chosen to efficiently recognize the waterline and output image coordinates. Finally, the internal and external parameters of the camera are calibrated. By applying the principles of camera imaging and installing surveillance cameras at the end of the tailings pond, the dry beach length is predicted. Experiments prove that this model can not only accurately predict the dry beach length but also has good stability in segmenting the water boundaries of different tailings ponds. It has a good effect on non-contact measurement in the field under real-time video stream mode, with an error controlled within 2%.
YANG J, SUN Y Q, SHEN TU N Y, et al. Research onVideo Measurement of the Leng th of TailingsPond Dry Beach Based on Mask R-CNN Algorithm [J]. Acta Metrologica Sinica, 2020, 41(12): 1468-1474.
[5]
LI Y, QI H, DAI J, et al. Fully convolutional instance-aware semantic segmentation [C]//IEEE Computer Society. Proceedings of the IEEE conference on computer vision and pattern recognition. Hawaii, USA, 2017: 2359-2367.
WU F H, CUI J X, ZHANG N, et al. The wheel surface defect detection based on the improvement of the YOLOV4 algorithm [J]. Acta Metrologica Sinica, 2022, 43(11): 1404-1411.
WU Z Q, MA B Y. Research on HEV Energy Distribution Strategy Based on Improved Deep Reinforcement Learning[J]. Acta Metrologica Sinica, 2023, 44(12): 1863-1871.
[16]
HOSSAIN M S, BETTS J M, PAPLINSKI A P. Dual Focal Loss to address class imbalance in semantic segmentation [J]. Neurocomputing, 2021, 462: 69-87.
XIE X Y, TIAN W Q, WANG Y H, et al. Analysis of the Safety Status and Management Countermeasures of Tailings Ponds in China [J]. Science and Technology of Safety Production in China, 2009, 5(2): 5-9.
LI W, GAO L. Road crack detection basedon improved watershed algorithm [J]. Computer Engineering and Application, 2013, 49(20): 263-266.
SONG K, YANG H, YIN Z. Multi-scale attention deep neural network for fast accurate object detection [J]. IEEE Transactions on Circuits and Systems for Video Technology, 2018, 29(10): 2972-2985.
[6]
CHEN H, SUN S. U-YOLO: higher precision YOLOv4 [C]//SPIE. Twelfth International Conference on Graphics and Image Processing(ICGIP 2020). GuiLin, China, 2021, 11720: 48-53.
REDMON J, DIVVALA S, GIRSHICK R, et al. You only look once: Unified, real-time object detection [C]//IEEE Computer Society. Proceedings of the IEEE conference on computer vision and pattern recognition. Las Vegas, USA, 2016: 779-788.
[10]
ZHOU Y, ZHU W, HE Y, et al. YOLOv8-based Spatial Target Part Recognition [C]//IEEE. 2023 IEEE 3rd International Conference on Information Technology, Big Data and Artificial Intelligence(ICIBA). Hong Kong, China, 2023, 3: 1684-1687.
[12]
ZHOU Q, LIU H, QIU Y, et al. Object Detection for Construction Waste Based on an Improved YOLOv5 Model [J]. Sustainability, 2022, 15(1): 681.
[14]
GLOROT X, BORDES A, BENGIO Y. Deep sparse rectifier neural networks [C]//JMLR Workshop and Conference Proceedings. Proceedings of the fourteenth international conference on artificial intelligence and statistics. Fort Lauderdale, USA, 2011: 315-323.
[15]
ZHENG Z, WANG P, LIU W, et al. Distance-IoU loss: Faster and better learning for bounding box regression [C]//AAAI. Proceedings of the AAAI conference on artificial intelligence. Paris, France, 2020, 34(7): 12993-13000.
[11]
ZHANG S, CHI C, YAO Y, et al. Bridging the gap between anchor-based and anchor-free detection via adaptive training sample selection [C]//CVPR. Proceedings of the IEEE/CVF conference on computer vision and pattern recognition. Seattle, USA, 2020: 9759-9768.
[13]
LI X, WANG W, WU L, et al. Generalized focal loss: Learning qualified and distributed bounding boxes for dense object detection [J]. Advances in Neural Information Processing Systems, 2020, 33: 21002-21012.
HUANG Z Q, SU Y, WANG Q W, et al. Research on External Reference Calibration Methods for 2D Lidar and Visible Light Cameras [J]. Chinese Journal of Scientific Instrument, 2020, 41(9): 121-129.