arrow
返回

Improved object detection via large kernel attention

delete2024-04-01
delete8
PRE
AI
Z
Zhaoxun Wang
李玉善 封面图
李玉善 (Yushan Li) *
刘
刘扬 (Yang Liu)
F
Fanyu Meng
DOI:10.1016/j.eswa.2023.122507delete
delete原文链接
delete原文求助
delete分享
delete收藏
摘要

摘要

En 中文
Accurate and real-time object detection is crucial for autonomous vehicles. For in-vehicle computing platforms, a giant model is difficult to achieve in real time and also increases the hardware cost. Moreover, a lightweight model built from a lot of depth-wise separable convolution layers cannot achieve the accuracy required for autonomous driving. We introduce Large Kernel Attention (LKA) technology to decouple the large kernel convolutions. It can combine high accuracy with small computational cost. Furthermore, we use LKA as the basis for designing a new module (Res-VAN) that can be used to build backbone networks. This study focuses on how the Res-VAN module can be deployed to improve the YOLOv5 in order to reduce the computational effort of the model, but maintain its accuracy. We named the model LKA-YOLO and validated it on PASCAL VOC dataset, MS COCO 2017 dataset and VirDrone2019 datasets. Experimental results show that LKA-YOLO reduces the computation of YOLOv5 by 57.5% (16 GFLOPs vs. 6.8 GFLOPs). And the feature extraction ability of LKA-YOLO is stronger than YOLOv5 on small datasets. The results on VirDrone2019 dataset show that our proposed method has a great advantage in dealing with tiny objects. Meanwhile, comparing with YOLO family on the COCO 2017 dataset, the metrics of LKA-YOLO are also outstanding.
Keyword:
Large kernel attention
Object detection
YOLOv5
Lightweight
Autonomous driving

期刊

Expert Systems with Applications 封面图
Expert Systems with Applications
IF:
7.5
论文数:
2.9W
被引数:
10.2W

机构

U
university of glasgow
学者数:
3.5W
论文数: 3.1W
被引数: 37
引用论文

引用论文

Enhancing Geometric Factors in Model Learning and Inference for Object Detection and Instance Segmentation
err2022-08-01
err620
errOAAI
errZheng, Zhaohui; Wang, Ping; Ren, Dongwei; Liu, Wei; Ye, Rongguang; Hu, Qinghua; Zuo, Wangmeng
err分享
err收藏
err分享
err收藏
Weighted Multi-view Clustering with Feature. Selection
err2016-05-01
err157
PREAI
errXu, Yu-Meng; Wang, Chang-Dong; Lai, Jian-Huang
err分享
err收藏
err分享
err收藏
学者 查看更多内容