Improved YOLOv8 track foreign obj ect detection based on lightweight convolution and information enhancement

被引:0
作者
Cuil, Songbo [1 ]
Zhang, Yong [1 ]
Cao, Fang [2 ]
Qu, Tao [3 ]
Sun, Xiaodong [4 ]
机构
[1] Univ Sci & Technol Liaoning, Sch Elect & Informat Engn, Anshan 114051, Peoples R China
[2] Northeast Univ Sci & Technol Grp, Shenyang 110004, Peoples R China
[3] Angang Grp Min Gongchangling Co Ltd, Power Branch, Liaoyang 111007, Peoples R China
[4] Anshan Steel Grp Min Co, Anshan 114046, Peoples R China
来源
2024 14TH ASIAN CONTROL CONFERENCE, ASCC 2024 | 2024年
关键词
YOLOv8; lightweight convolution; information enhancement; foreign object detection; small target detection;
D O I
暂无
中图分类号
TP [自动化技术、计算机技术];
学科分类号
0812 ;
摘要
Railroads are among the most common modes of transportation globally, but their tracks are typically open, making them vulnerable to the sudden and random appearance of foreign objects. Consequently, railroad accidents resulting from these obstacles have become increasingly common in recent years. This paper aims to efficiently and rapidly detect foreign objects on tracks to safeguard lives and property. Firstly, to reduce the model size, we introduce a lightweight convolutional module, AMConv. Additionally, to enhance the distinction between background and foreign objects, we propose a Contextual Information Enhancement Module (ClEM) in the backbone network to extract more pertinent feature information about foreign objects from the original image. Lastly, to enhance the accuracy of detecting foreign objects with small profiles, we remove the largest detection head and instead incorporate a smaller-sized detection head. Through numerous experiments conducted on the dataset, alongside ablation and comparison experiments with state-of-the-art models, the effectiveness of our proposed model is demonstrated.
引用
收藏
页码:1260 / 1265
页数:6
相关论文
共 20 条
[1]  
Bochkovskiy A, 2020, Arxiv, DOI [arXiv:2004.10934, DOI 10.48550/ARXIV.2004.10934, 10.48550/arXiv.2004.10934]
[2]  
Chen W., 2022, Computational Intelligence and Neuroscience, V2022
[3]  
Du X. Y., 2022, IEEE INT C ADV INF T, P87
[4]   CenterNet: Keypoint Triplets for Object Detection [J].
Duan, Kaiwen ;
Bai, Song ;
Xie, Lingxi ;
Qi, Honggang ;
Huang, Qingming ;
Tian, Qi .
2019 IEEE/CVF INTERNATIONAL CONFERENCE ON COMPUTER VISION (ICCV 2019), 2019, :6568-6577
[5]   Rich feature hierarchies for accurate object detection and semantic segmentation [J].
Girshick, Ross ;
Donahue, Jeff ;
Darrell, Trevor ;
Malik, Jitendra .
2014 IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2014, :580-587
[6]   GhostNet: More Features from Cheap Operations [J].
Han, Kai ;
Wang, Yunhe ;
Tian, Qi ;
Guo, Jianyuan ;
Xu, Chunjing ;
Xu, Chang .
2020 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2020, :1577-1586
[7]   Improved Mask R-CNN for obstacle detection of rail transit [J].
He, Deqiang ;
Qiu, Yefeng ;
Miao, Jian ;
Zou, Zhiheng ;
Li, Kai ;
Ren, Chonghui ;
Shen, Guoqiang .
MEASUREMENT, 2022, 190
[8]   Rail Transit Obstacle Detection Based on Improved CNN [J].
He, Deqiang ;
Zou, Zhiheng ;
Chen, Yanjun ;
Liu, Bin ;
Miao, Jian .
IEEE TRANSACTIONS ON INSTRUMENTATION AND MEASUREMENT, 2021, 70
[9]  
He KM, 2017, IEEE I CONF COMP VIS, P2980, DOI [10.1109/TPAMI.2018.2844175, 10.1109/ICCV.2017.322]
[10]  
Jiang CS, 2022, ASIA CONTROL CONF AS, P1084, DOI 10.23919/ASCC56756.2022.9828211