计算机科学
人工智能
目标检测
块(置换群论)
搜救
特征(语言学)
计算机视觉
对象(语法)
模式识别(心理学)
数据挖掘
机器人
几何学
数学
语言学
哲学
作者
Xiaotian Wang,Zhizhong Pan,Ningxin He,Tiegang Gao
出处
期刊:Journal of Intelligent and Fuzzy Systems
[IOS Press]
日期:2023-08-24
卷期号:45 (3): 3575-3586
摘要
Unmanned aerial vehicles (UAVs) play a crucial role in maritime search and rescue missions, capturing images of open water scenarios and assisting in object detection. Previous object detection models have mainly focused on general scenarios. However, existing object detection models have mainly focused on general scenarios, while images captured by UAVs in vast ocean scenarios often contain numerous small objects that significantly degrade the performance of the original models. To address this challenge, we propose a model that can automatically detect objects in images captured by UAVs during maritime search and rescue missions. Our approach involves designing a new detection head with higher resolution feature maps and more comprehensive feature information to improve the detection of small objects. Additionally, we integrate Swin Transformer blocks into the small object detection head, which can improve the model’s ability to obtain abundant contextual information and thus improves the model’s ability to detect small objects. Moreover, we fuse the Convolutional Block Attention Model into the small object detection head to help the model focus on important features. Finally, we adopt a model ensemble strategy to further improve the mean average precision (mAP). Our proposed model achieves a 4.05% improvement in mAP compared to the baseline model. Furthermore, our model outperforms the previous state-of-the-art model on the SeaDronesSee dataset in terms of fewer parameters, lower training costs, and higher mAP.
科研通智能强力驱动
Strongly Powered by AbleSci AI