Cargando…

DB-YOLO: A Duplicate Bilateral YOLO Network for Multi-Scale Ship Detection in SAR Images

With the wide application of convolutional neural networks (CNNs), a variety of ship detection methods based on CNNs in synthetic aperture radar (SAR) images were proposed, but there are still two main challenges: (1) Ship detection requires high real-time performance, and a certain detection speed...

Descripción completa

Detalles Bibliográficos
Autores principales: Zhu, Haozhen, Xie, Yao, Huang, Huihui, Jing, Chen, Rong, Yingjiao, Wang, Changyuan
Formato: Online Artículo Texto
Lenguaje:English
Publicado: MDPI 2021
Materias:
Acceso en línea:https://www.ncbi.nlm.nih.gov/pmc/articles/PMC8662457/
https://www.ncbi.nlm.nih.gov/pubmed/34884163
http://dx.doi.org/10.3390/s21238146
_version_ 1784613441189707776
author Zhu, Haozhen
Xie, Yao
Huang, Huihui
Jing, Chen
Rong, Yingjiao
Wang, Changyuan
author_facet Zhu, Haozhen
Xie, Yao
Huang, Huihui
Jing, Chen
Rong, Yingjiao
Wang, Changyuan
author_sort Zhu, Haozhen
collection PubMed
description With the wide application of convolutional neural networks (CNNs), a variety of ship detection methods based on CNNs in synthetic aperture radar (SAR) images were proposed, but there are still two main challenges: (1) Ship detection requires high real-time performance, and a certain detection speed should be ensured while improving accuracy; (2) The diversity of ships in SAR images requires more powerful multi-scale detectors. To address these issues, a SAR ship detector called Duplicate Bilateral YOLO (DB-YOLO) is proposed in this paper, which is composed of a Feature Extraction Network (FEN), Duplicate Bilateral Feature Pyramid Network (DB-FPN) and Detection Network (DN). Firstly, a single-stage network is used to meet the need of real-time detection, and the cross stage partial (CSP) block is used to reduce the redundant parameters. Secondly, DB-FPN is designed to enhance the fusion of semantic and spatial information. In view of the ships in SAR image are mainly distributed with small-scale targets, the distribution of parameters and computation values between FEN and DB-FPN in different feature layers is redistributed to solve the multi-scale detection. Finally, the bounding boxes and confidence scores are given through the detection head of YOLO. In order to evaluate the effectiveness and robustness of DB-YOLO, comparative experiments with the other six state-of-the-art methods (Faster R-CNN, Cascade R-CNN, Libra R-CNN, FCOS, CenterNet and YOLOv5s) on two SAR ship datasets, i.e., SSDD and HRSID, are performed. The experimental results show that the AP50 of DB-YOLO reaches 97.8% on SSDD and 94.4% on HRSID, respectively. DB-YOLO meets the requirement of real-time detection (48.1 FPS) and is superior to other methods in the experiments.
format Online
Article
Text
id pubmed-8662457
institution National Center for Biotechnology Information
language English
publishDate 2021
publisher MDPI
record_format MEDLINE/PubMed
spelling pubmed-86624572021-12-11 DB-YOLO: A Duplicate Bilateral YOLO Network for Multi-Scale Ship Detection in SAR Images Zhu, Haozhen Xie, Yao Huang, Huihui Jing, Chen Rong, Yingjiao Wang, Changyuan Sensors (Basel) Article With the wide application of convolutional neural networks (CNNs), a variety of ship detection methods based on CNNs in synthetic aperture radar (SAR) images were proposed, but there are still two main challenges: (1) Ship detection requires high real-time performance, and a certain detection speed should be ensured while improving accuracy; (2) The diversity of ships in SAR images requires more powerful multi-scale detectors. To address these issues, a SAR ship detector called Duplicate Bilateral YOLO (DB-YOLO) is proposed in this paper, which is composed of a Feature Extraction Network (FEN), Duplicate Bilateral Feature Pyramid Network (DB-FPN) and Detection Network (DN). Firstly, a single-stage network is used to meet the need of real-time detection, and the cross stage partial (CSP) block is used to reduce the redundant parameters. Secondly, DB-FPN is designed to enhance the fusion of semantic and spatial information. In view of the ships in SAR image are mainly distributed with small-scale targets, the distribution of parameters and computation values between FEN and DB-FPN in different feature layers is redistributed to solve the multi-scale detection. Finally, the bounding boxes and confidence scores are given through the detection head of YOLO. In order to evaluate the effectiveness and robustness of DB-YOLO, comparative experiments with the other six state-of-the-art methods (Faster R-CNN, Cascade R-CNN, Libra R-CNN, FCOS, CenterNet and YOLOv5s) on two SAR ship datasets, i.e., SSDD and HRSID, are performed. The experimental results show that the AP50 of DB-YOLO reaches 97.8% on SSDD and 94.4% on HRSID, respectively. DB-YOLO meets the requirement of real-time detection (48.1 FPS) and is superior to other methods in the experiments. MDPI 2021-12-06 /pmc/articles/PMC8662457/ /pubmed/34884163 http://dx.doi.org/10.3390/s21238146 Text en © 2021 by the authors. https://creativecommons.org/licenses/by/4.0/Licensee MDPI, Basel, Switzerland. This article is an open access article distributed under the terms and conditions of the Creative Commons Attribution (CC BY) license (https://creativecommons.org/licenses/by/4.0/).
spellingShingle Article
Zhu, Haozhen
Xie, Yao
Huang, Huihui
Jing, Chen
Rong, Yingjiao
Wang, Changyuan
DB-YOLO: A Duplicate Bilateral YOLO Network for Multi-Scale Ship Detection in SAR Images
title DB-YOLO: A Duplicate Bilateral YOLO Network for Multi-Scale Ship Detection in SAR Images
title_full DB-YOLO: A Duplicate Bilateral YOLO Network for Multi-Scale Ship Detection in SAR Images
title_fullStr DB-YOLO: A Duplicate Bilateral YOLO Network for Multi-Scale Ship Detection in SAR Images
title_full_unstemmed DB-YOLO: A Duplicate Bilateral YOLO Network for Multi-Scale Ship Detection in SAR Images
title_short DB-YOLO: A Duplicate Bilateral YOLO Network for Multi-Scale Ship Detection in SAR Images
title_sort db-yolo: a duplicate bilateral yolo network for multi-scale ship detection in sar images
topic Article
url https://www.ncbi.nlm.nih.gov/pmc/articles/PMC8662457/
https://www.ncbi.nlm.nih.gov/pubmed/34884163
http://dx.doi.org/10.3390/s21238146
work_keys_str_mv AT zhuhaozhen dbyoloaduplicatebilateralyolonetworkformultiscaleshipdetectioninsarimages
AT xieyao dbyoloaduplicatebilateralyolonetworkformultiscaleshipdetectioninsarimages
AT huanghuihui dbyoloaduplicatebilateralyolonetworkformultiscaleshipdetectioninsarimages
AT jingchen dbyoloaduplicatebilateralyolonetworkformultiscaleshipdetectioninsarimages
AT rongyingjiao dbyoloaduplicatebilateralyolonetworkformultiscaleshipdetectioninsarimages
AT wangchangyuan dbyoloaduplicatebilateralyolonetworkformultiscaleshipdetectioninsarimages