基于改进BiFPN的YOLOv4增强型遥感目标检测

IF 3.7 3区 地球科学 Q2 ENVIRONMENTAL SCIENCES Egyptian Journal of Remote Sensing and Space Sciences Pub Date : 2023-08-01 DOI:10.1016/j.ejrs.2023.04.003
Fuzhen Zhu, Yuying Wang, Jingyi Cui, Guoxin Liu, Huiling Li
{"title":"基于改进BiFPN的YOLOv4增强型遥感目标检测","authors":"Fuzhen Zhu,&nbsp;Yuying Wang,&nbsp;Jingyi Cui,&nbsp;Guoxin Liu,&nbsp;Huiling Li","doi":"10.1016/j.ejrs.2023.04.003","DOIUrl":null,"url":null,"abstract":"<div><p>To solve problems for false detection, inadequate regression performance of anchor frames, and the inability to detect small targets in traditional multiscale target detection methods based on YOLOv4, we propose a novel target detection framework named as Enhanced YOLOv4. Firstly, our improved BiFPN replaced the original PANet as the feature fusion module, which can achieve multi-scale feature fusion by way of shared weights. Secondly, the channel attention mechanism (CAM) was embedded before the detection head to highlight the correlation between channels so that small targets can be get more attention. At last, to improve the anchor box regression effect and accelerate the training speed of YOLOv4, we improved the net training loss function, in which the original CIoU was replaced by CDIoU. The experimental results on the DOTA dataset validate our improvement. The mAP of our method is 90.88%, and the frame rate reached 58.76 FPS, at the same time, the speed of detection is not affected significantly.</p></div>","PeriodicalId":48539,"journal":{"name":"Egyptian Journal of Remote Sensing and Space Sciences","volume":"26 2","pages":"Pages 351-360"},"PeriodicalIF":3.7000,"publicationDate":"2023-08-01","publicationTypes":"Journal Article","fieldsOfStudy":null,"isOpenAccess":false,"openAccessPdf":"","citationCount":"5","resultStr":"{\"title\":\"Target detection for remote sensing based on the enhanced YOLOv4 with improved BiFPN\",\"authors\":\"Fuzhen Zhu,&nbsp;Yuying Wang,&nbsp;Jingyi Cui,&nbsp;Guoxin Liu,&nbsp;Huiling Li\",\"doi\":\"10.1016/j.ejrs.2023.04.003\",\"DOIUrl\":null,\"url\":null,\"abstract\":\"<div><p>To solve problems for false detection, inadequate regression performance of anchor frames, and the inability to detect small targets in traditional multiscale target detection methods based on YOLOv4, we propose a novel target detection framework named as Enhanced YOLOv4. Firstly, our improved BiFPN replaced the original PANet as the feature fusion module, which can achieve multi-scale feature fusion by way of shared weights. Secondly, the channel attention mechanism (CAM) was embedded before the detection head to highlight the correlation between channels so that small targets can be get more attention. At last, to improve the anchor box regression effect and accelerate the training speed of YOLOv4, we improved the net training loss function, in which the original CIoU was replaced by CDIoU. The experimental results on the DOTA dataset validate our improvement. The mAP of our method is 90.88%, and the frame rate reached 58.76 FPS, at the same time, the speed of detection is not affected significantly.</p></div>\",\"PeriodicalId\":48539,\"journal\":{\"name\":\"Egyptian Journal of Remote Sensing and Space Sciences\",\"volume\":\"26 2\",\"pages\":\"Pages 351-360\"},\"PeriodicalIF\":3.7000,\"publicationDate\":\"2023-08-01\",\"publicationTypes\":\"Journal Article\",\"fieldsOfStudy\":null,\"isOpenAccess\":false,\"openAccessPdf\":\"\",\"citationCount\":\"5\",\"resultStr\":null,\"platform\":\"Semanticscholar\",\"paperid\":null,\"PeriodicalName\":\"Egyptian Journal of Remote Sensing and Space Sciences\",\"FirstCategoryId\":\"89\",\"ListUrlMain\":\"https://www.sciencedirect.com/science/article/pii/S1110982323000224\",\"RegionNum\":3,\"RegionCategory\":\"地球科学\",\"ArticlePicture\":[],\"TitleCN\":null,\"AbstractTextCN\":null,\"PMCID\":null,\"EPubDate\":\"\",\"PubModel\":\"\",\"JCR\":\"Q2\",\"JCRName\":\"ENVIRONMENTAL SCIENCES\",\"Score\":null,\"Total\":0}","platform":"Semanticscholar","paperid":null,"PeriodicalName":"Egyptian Journal of Remote Sensing and Space Sciences","FirstCategoryId":"89","ListUrlMain":"https://www.sciencedirect.com/science/article/pii/S1110982323000224","RegionNum":3,"RegionCategory":"地球科学","ArticlePicture":[],"TitleCN":null,"AbstractTextCN":null,"PMCID":null,"EPubDate":"","PubModel":"","JCR":"Q2","JCRName":"ENVIRONMENTAL SCIENCES","Score":null,"Total":0}
引用次数: 5

摘要

为了解决传统基于YOLOv4的多尺度目标检测方法中存在的错误检测、锚帧回归性能不足以及无法检测小目标的问题,我们提出了一种新的目标检测框架Enhanced YOLOv4。首先,我们改进的BiFPN取代了原来的PANet作为特征融合模块,可以通过共享权重的方式实现多尺度特征融合。其次,在检测头之前嵌入了通道注意机制(CAM),以突出通道之间的相关性,从而使小目标能够得到更多的关注。最后,为了提高锚盒回归效果,加快YOLOv4的训练速度,我们改进了净训练损失函数,将原来的CIoU替换为CDIoU。DOTA数据集的实验结果验证了我们的改进。我们的方法的mAP为90.88%,帧率达到58.76 FPS,同时检测速度没有受到显著影响。
本文章由计算机程序翻译,如有差异,请以英文原文为准。
查看原文
分享 分享
微信好友 朋友圈 QQ好友 复制链接
本刊更多论文
Target detection for remote sensing based on the enhanced YOLOv4 with improved BiFPN

To solve problems for false detection, inadequate regression performance of anchor frames, and the inability to detect small targets in traditional multiscale target detection methods based on YOLOv4, we propose a novel target detection framework named as Enhanced YOLOv4. Firstly, our improved BiFPN replaced the original PANet as the feature fusion module, which can achieve multi-scale feature fusion by way of shared weights. Secondly, the channel attention mechanism (CAM) was embedded before the detection head to highlight the correlation between channels so that small targets can be get more attention. At last, to improve the anchor box regression effect and accelerate the training speed of YOLOv4, we improved the net training loss function, in which the original CIoU was replaced by CDIoU. The experimental results on the DOTA dataset validate our improvement. The mAP of our method is 90.88%, and the frame rate reached 58.76 FPS, at the same time, the speed of detection is not affected significantly.

求助全文
通过发布文献求助,成功后即可免费获取论文全文。 去求助
来源期刊
CiteScore
8.10
自引率
0.00%
发文量
85
审稿时长
48 weeks
期刊介绍: The Egyptian Journal of Remote Sensing and Space Sciences (EJRS) encompasses a comprehensive range of topics within Remote Sensing, Geographic Information Systems (GIS), planetary geology, and space technology development, including theories, applications, and modeling. EJRS aims to disseminate high-quality, peer-reviewed research focusing on the advancement of remote sensing and GIS technologies and their practical applications for effective planning, sustainable development, and environmental resource conservation. The journal particularly welcomes innovative papers with broad scientific appeal.
期刊最新文献
Editorial Board Spectral–Spatial Adaptive Weighted Fusion and Residual Dense Network for hyperspectral image classification New radio-seismic indicator for ELF seismic precursors detectability Estimation of above ground biomass of mangrove forest plot using terrestrial laser scanner Efficient bundle optimization for accurate camera pose estimation in mobile augmented reality systems
×
引用
GB/T 7714-2015
复制
MLA
复制
APA
复制
导出至
BibTeX EndNote RefMan NoteFirst NoteExpress
×
×
提示
您的信息不完整,为了账户安全,请先补充。
现在去补充
×
提示
您因"违规操作"
具体请查看互助需知
我知道了
×
提示
现在去查看 取消
×
提示
确定
0
微信
客服QQ
Book学术公众号 扫码关注我们
反馈
×
意见反馈
请填写您的意见或建议
请填写您的手机或邮箱
已复制链接
已复制链接
快去分享给好友吧!
我知道了
×
扫码分享
扫码分享
Book学术官方微信
Book学术文献互助
Book学术文献互助群
群 号:481959085
Book学术
文献互助 智能选刊 最新文献 互助须知 联系我们:info@booksci.cn
Book学术提供免费学术资源搜索服务,方便国内外学者检索中英文文献。致力于提供最便捷和优质的服务体验。
Copyright © 2023 Book学术 All rights reserved.
ghs 京公网安备 11010802042870号 京ICP备2023020795号-1