Fuzhen Zhu, Yuying Wang, Jingyi Cui, Guoxin Liu, Huiling Li
{"title":"基于改进BiFPN的YOLOv4增强型遥感目标检测","authors":"Fuzhen Zhu, Yuying Wang, Jingyi Cui, Guoxin Liu, 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, Yuying Wang, Jingyi Cui, Guoxin Liu, 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}
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.
期刊介绍:
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.