计算机科学
交叉口(航空)
断层(地质)
故障检测与隔离
人工智能
目标检测
特征(语言学)
块(置换群论)
干扰(通信)
GSM演进的增强数据速率
图像融合
计算机视觉
实时计算
图像(数学)
频道(广播)
模式识别(心理学)
工程类
数学
航空航天工程
哲学
几何学
地震学
执行机构
地质学
语言学
计算机网络
作者
Longxin Zhang,Jingsheng Chen,Wenliang Zeng,Zhao Tong,Zhiqiang Wen,Xiaoyun Zhang
出处
期刊:IEEE Sensors Journal
[Institute of Electrical and Electronics Engineers]
日期:2023-10-15
卷期号:23 (20): 25105-25115
被引量:2
标识
DOI:10.1109/jsen.2023.3313155
摘要
The current train image fault detection model does not consider detection accuracy and speed, and its parameters and calculation amount are large. Consequently, the model cannot easily satisfy the requirements of the actual edge computing scenario. Therefore, this study proposes a lightweight train image fault detection network (FDNet) under an edge computing environment. FDNet makes three improvements based on YOLOX. First, FDNet designs dual-path enhanced channel attention (DECA) module and fuses the path aggregation network (PAN) to form a DECA-PAN module to suppress background interference, highlight the object, and improve the feature fusion ability of the model. Second, focal loss and alpha-complete intersection over union (Alpha-CIoU) loss are used to optimize the loss function to alleviate the problem of small image samples and uneven fault categories. Third, the receptive field fusion block is added to the FDNet detection head, and three groups of detection heads are devised to detect objects at different scales to improve the detection ability of the model on multiscale objects. Experimental results on the train image fault dataset demonstrate that the mean average precision (mAP) and speed of FDNet reach 94.82% and 103.4 frames per second (FPS), respectively, whereas its parameters and floating-point operations are less than one-sixth those of the well-known YOLOv4. The experimental results of FDNet, YOLOv5-s, YOLOX-s, YOLOv6-T, and YOLOv7-tiny in the public dataset COCO2017 show that FDNet achieves the highest mAP of 60.60% while maintaining a high detection speed among all lightweight models. The noise and illumination robustness tests show that FDNet has good anti-interference ability.
科研通智能强力驱动
Strongly Powered by AbleSci AI