Citation: | WANG Manli,YANG Shuang,ZHANG Changsen. An improved yolov8n based dislocation detection algorithm for shaft rigid tank channel joint[J]. Coal Science and Technology,2024,52(S2):236−248. DOI: 10.12438/cst.2023-0788 |
As the guiding device of the hoisting system, the shaft rigid tank channel is the basis for the safe and stable operation of the hoisting system, which is prone to lateral displacement and deformation during the use process, and will cause strong vibration of the hoisting system in the event of unevenness, which seriously affects the safety of the operation of the hoisting system. In order to timely discover the faults of vertical well rigid tank channel joints and eliminate the hidden dangers of the hoisting system operation, a vertical well rigid tank channel joint misalignment detection algorithm YOLOv8n-CFW is proposed to improve YOLOv8n. First of all, for the blurred wellbore, darkness, strong light and the existence of complex background imaging environment, the YOLOv8n is insufficient to extract the image features of the rigid tank channel of the vertical well, and it incorporates the CBAM Attention Mechanism Module, which combines the channel attention mechanism with the spatial attention mechanism to form a new convolutional block structure for better feature fusion and helps the model to focus more on the important parts of the input image, thus improving the model's recognition accuracy and generalization ability, and overcoming the problem of YOLOv8n's trunk network's lack of feature extraction ability and generalization ability for locally important information under blurred, dark, and bright light environments. feature extraction ability is insufficient and the generalization ability is poor; then, to lighten the network, Faster_Block is used instead of Bottleneck to reduce the computational complexity of the C2f module, which overcomes the shortcomings of YOLOv8n network that the amount of model parameters is too large to be deployed with difficulty; and then, to inhibit the harmful gradients generated by low-quality images, the WIoU loss function is used instead of the CIoU to improve the anchor frame prediction accuracy, and the WIoU v3 loss function, which suppresses the harmful gradients generated by low-quality images by dynamically distributing the gradient gain, overcomes the insufficient localization performance of the YOLOv8n network, so that the network locates more accurately to the edge of the rigid tank channel in the joint mismatch detection, thus reducing the error, and further improves the model recognition accuracy and generalization ability; finally, the inference process in the Finally, the region of interest is set in the inference process to further suppress the background image interference, and the non-fixed focus ranging method is used to calculate the tank channel joint offset size. The experiments show that compared with the basic YOLOv8n network, the YOLOv8n-CFW detection network improves the precision P by 1.4%, the recall R by 8.2%, the average precision mAP by 6.5% from 83.8% in YOLOv8n to 90.3%, and the model size is reduced by 1.4MB compared with other YOLO algorithms in the vertical wells with rigid tank channel joints dataset. YOLO algorithms, YOLOv8n-CFW has a significant advantage in misalignment detection of rigid tank channel joints in vertical wells.
[1] |
LI H,MA Y,BAO H,et al. Probabilistic hough transform for rectifying industrial nameplate images:a novel strategy for improved text detection and precision in difficult environments[J]. Applied sciences-basel,2023,13(7):1−11.
|
[2] |
BO Wu,WEI Li,FAN Jiang. Fault diagnosis of mine shaft guide rails using vibration signal analysis based on dynamic time warping[J]. Symmetry,2018,10(10),Article ID:500.
|
[3] |
马天兵,吴强,王鑫泉等. 基于机器视觉与激光融合的刚性罐道故障定位技术[J]. 煤矿安全,2020,51(1):134−137.
MA Tianbing,WU Qiang,WANG Xinquan,et al. Fault location technology of rigid tank passage based on machine vision and laser fusion[J]. Safety in Coal Mine,2020,51(1):134−137.
|
[4] |
李爱民,包从望. 基于EMD-SVM的刚性罐道故障诊断研究[J]. 煤炭技术,2016,35(4):264−266.
LI Aimin,BAO Congwang. Research on Fault Diagnosis of Rigid tank Passage Based on EMD-SVM[J]. Coal Technology,2016,35(4):264−266.
|
[5] |
马天兵,王孝东,杜菲等. 基于GA-SVM的刚性罐道故障诊断[J]. 工程设计学报,2019,26(2):170−176, 222.
MA Tianbing,WANG Xiaodong,DU Fei,et al. Fault diagnosis of rigid tank track based on GA-SVM[J]. Chinese Journal of Engineering Design,2019,26(2):170−176, 222.
|
[6] |
马天兵,王鑫泉,王孝东. 基于EMD-PNN网络的刚性罐道故障诊断方法[J]. 电子测量与仪器学报,2019,33(3):58−64.
MA Tianbing,WANG Xinquan,WANG Xiaodong. Fault diagnosis method for rigid tank channel based on EMD-PNN network[J]. Journal of Electronic Measurement and Instrument,2019,33(3):58−64.
|
[7] |
马天兵,王孝东,杜菲等. 基于小波包和BP神经网络的刚性罐道故障诊断[J]. 工矿自动化,2018,44(8):76−80.
MA Tianbing,WANG Xiaodong,DU Fei,et al. Fault diagnosis of rigid tank passage based on wavelet packet and BP neural network[J]. Industry and Mine Automation,2018,44(8):76−80.
|
[8] |
REDMON J,DIVVALA S,GIRSHICK R,et al. You only look once:unified,real-time object detection [C]// Proceedings of 2016 IEEE Conference on Computer Vision and Pattern Recognition. Las Vegas:IEEE,2016:779−788.
|
[9] |
AHMAD T,MA Y,YAHYA M,et al. Object detection through modified YOLO neural network[J]. scientific programming. 2020,12(2). DOI:10.1155/ 2020/ 8403262
|
[10] |
REDMON J,FARHADI A. YOLO9000:better,faster,stronger [C]//Proceedings of 2017 IEEE Conference on Computer Vision and Pattern Recognition. Honolulu:IEEE,2017:6517−6525
|
[11] |
HE Z,MA Y,WANG Z,et al. A novel efficient method for welding spots detection[J]. Multimedia tools and applications,2022,81(18):26381−26401.
|
[12] |
LI X,SHI B,NIE T,et al. Multi-object recognition method based on improved YOLOv2 model[J]. Information technology and control,2021,50(1):13−27.
|
[13] |
REDMON J,FARHADI A. YOLOv3:an incremental improvement [J]. arXiv,2018:1804. 02767.
|
[14] |
FANGRONG Zhou,HAO Pan,GUOCHAO Qian,et al. Insulator and burst fault detection using an improved Yolov3 algorithm[J]. Journal of sensors,2022,17(1):1−12.
|
[15] |
TENG Shuai,LIU Zongchao,LI Xiaoda. Improved YOLOv3-based bridge surface defect detection by combining high- and low-resolution feature images[J]. Buildings,2022,12(8):1−9.
|
[16] |
SHI Tingchao,LIU Mingyong,NIU Yun,et al. Underwater targets detection and classification in complex scenes based on an improved YOLOv3 algorithm[J]. Journal of electronic imaging,2020,29(4):1−11.
|
[17] |
WANG Zhihui,ZHU Houying,JIA Xianqing,et al. Surface defect detection with modified real-time detector YOLOv3[J]. Journal of sensors,2022,9(2):1−12.
|
[18] |
ZHU X K,LYU S C,WANG X,et al. TPH-YOLOv5:improved YOLOv5 based on transformer prediction head for object detection on drone-captured scenarios [C]//Proceedings of 2021 IEEE/CVF International Conference on Computer Vision Workshops. Montreal:IEEE,2021:2778-2788
|
[19] |
YANG Rujin,LI Wenfa,SHUANG Xinna,et al. KPE-YOLOv5:an improved small target detection algorithm based on YOLOv5[J]. Electronics,2023,12(4):1−9.
|
[20] |
MENG Xiaoliang,LUO Sen,WANG Weiling,et al. A detection model for corner cracks of continuous casting strand based on deep learning[J]. Ironmaking & steelmaking,2022,49(10):1048−1056.
|
[21] |
HONG Weiwei,MA Zenghong,YE Bingliang,et al. Detection of green asparagus in complex environments based on the improved YOLOv5 algorithm[J]. Sensors,2023,23(3):1−10. doi: 10.1109/JSEN.2023.3237045
|
[22] |
LIU Weimin,XIAO Yao,ZHENG Aiyun,et al. Research on fault diagnosis of steel surface based on improved YOLOV5[J]. Processes,2022,10(11):1−10.
|
[23] |
WANG C,BOCHKOVSKIY A,LIAO H M. YOLOv7:trainable bag-of-freebies sets new state-of-the-art for real-time object detectors[C]//2023 IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR),Vancouver,BC,Canada,2023,pp. 7464−7475.
|
[24] |
WU Delin,JIANG Shan,ZHAO Enlong,et al. Detection of camellia oleifera fruit in complex scenes by using YOLOv7 and data augmentation[J]. Applied sciences-basel,2022,12(22):1−10.
|
[25] |
LI Shuangbao,YU Jingyi,WANG Hao. Damages detection of aeroengine blades via deep learning algorithms[J]. IEEE Transactions on Instrumentation and Measurement,2023,72:1−11.
|
[26] |
ZHENG Jianfeng,WU Hang,ZHANG Han,et al. Insulator-defect detection algorithm based on improved YOLOv7[J]. Sensors,2022,22(22):1−10. doi: 10.1109/JSEN.2022.3218439
|
[27] |
WANG C Y ,YEH I H ,LIAO H Y . You only learn one representation:unified network for multiple tasks[J]. arXiv:2105.04206
|
[28] |
WOO S ,PARK J ,LEE J Y,et al. CBAM:Convolutional block attention module[J]. Springer,cham,2018,9(1):1−11.
|
[29] |
CHEN J,KAO S,HE H,et al. Run,don't walk:chasing higher FLOPS for faster neural networks[C]//2023 IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR),Vancouver,BC,Canada,2023,pp. 12021−12031.
|
[30] |
TONG Z,CHEN Y,XU Z,et al. Wise-IoU:Bounding box regression loss with dynamic focusing mechanism[J]. arXiv:2301.10051.
|
[31] |
ZHU J Y ,PARK T ,ISOLA P ,et al. Unpaired image-to-image translation using cycle-consistent adversarial network[C]//2017 IEEE International Conference on Computer Vision (ICCV),Venice,Italy,2017,pp. 2242−2251.
|
[32] |
ZHENG Z H,WANG P,LIU W,et al. Distance-IoU loss:faster and better learning for bounding box regression [C]// Proceedings of the 34th AAAI Conference on Artificial Intelligence. New York:AAAI,2020:12993−13000.
|