Improved YOLOv5 based on attention mechanism and FasterNet for foreign object detection on railway and airway tracks
编号:67 访问权限:仅限参会人 更新:2024-10-08 17:43:46 浏览:384次 张贴报告

报告开始:2024年10月25日 14:25(Asia/Bangkok)

报告时间:5min

所在会场:[PS] Poster Session [PS] Poster

暂无文件

摘要
In recent years, there have been frequent incidents of foreign objects intruding into railway and Airport runways. These objects can include pedestrians, vehicles, animals, and debris. This paper introduces an improved YOLOv5 architecture incorporating FasterNet and attention mechanisms to enhance the detection of foreign objects on railways and Airport runways. This study proposes a new dataset, AARFOD (Aero and Rail Foreign Object Detection), which combines two public datasets for detecting foreign objects in aviation and railway systems. The dataset aims to improve the recognition capabilities of foreign object targets. Experimental results on this large dataset have demonstrated significant performance improvements of the proposed model over the baseline YOLOv5 model, reducing computational requirements. improved YOLO model shows a significant improvement in precision by 1.2%, recall rate by 1.0%, and mAP@.5 by 0.6%, while mAP@.5-.95 remained unchanged. The parameters were reduced by approximately 25.12%, and GFLOPs were reduced by about 10.63%. In the ablation experiment, it is found that the FasterNet module can significantly reduce the number of parameters of the model, and the reference of the attention mechanism can slow down the performance loss caused by lightweight.
 
关键词
YOLOv5,FasterNet,NAM,foreign object detection
报告人
Zongqing Qi
Computer Science Computer Science, Stevens Institute of Technology, Hoboken NJ, U.S

稿件作者
Zongqing Qi Computer Science, Beijing University of Technology, Beijing, China
Danqing Ma Computer Information Technology, Northern Arizona University, Flagstaff, AZ, U.S
Jingyu Xu Computer Information Technology, Northern Arizona University, Arizona, U.S
Ao Xiang Digital Media Technology, University of Electronic Science and Technology of China, Sichuan, China
Hedi Qu Computer Science, Shenzhen SmartChip Microelectronics Technology Co., Ltd., China
发表评论
验证码 看不清楚,更换一张
全部评论
重要日期
  • 会议日期

    10月24日

    2024

    10月27日

    2024

  • 10月14日 2024

    初稿截稿日期

  • 10月29日 2024

    注册截止日期

  • 10月31日 2024

    报告提交截止日期

主办单位
国际科学联合会
IEEE泰国分会
IEEE计算机学会泰国分会
历届会议
移动端
在手机上打开
小程序
打开微信小程序
客服
扫码或点此咨询