ADSSD: Improved Single-Shot Detector with Attention Mechanism and Dilated Convolution
The detection of small objects is easily affected by background information, and a lack of context information makes detection difficult. Therefore, small object detection has become an extremely challenging task. Based on the above problems, we proposed a Single-Shot MultiBox Detector with an atten...
| Main Authors: | , , |
|---|---|
| Format: | Article |
| Language: | English |
| Published: |
MDPI AG
2023-03-01
|
| Series: | Applied Sciences |
| Subjects: | |
| Online Access: | https://www.mdpi.com/2076-3417/13/6/4038 |
| _version_ | 1797613534116315136 |
|---|---|
| author | Jian Ni Rui Wang Jing Tang |
| author_facet | Jian Ni Rui Wang Jing Tang |
| author_sort | Jian Ni |
| collection | DOAJ |
| description | The detection of small objects is easily affected by background information, and a lack of context information makes detection difficult. Therefore, small object detection has become an extremely challenging task. Based on the above problems, we proposed a Single-Shot MultiBox Detector with an attention mechanism and dilated convolution (ADSSD). In the attention module, we strengthened the connection between information in space and channels while using cross-layer connections to accelerate training. In the multi-branch dilated convolution module, we combined three expansion convolutions with different dilated ratios to obtain multi-scale context information and used hierarchical feature fusion to reduce the gridding effect. The results show that on PASCAL VOC2007 and VOC2012 datasets, our 300 × 300 input ADSSD model reaches 78.4% mAP and 76.1% mAP. The results outperform those of SSD and other advanced detectors; the effect of some small object detection is significantly improved. Moreover, the performance of the ADSSD in object detection affected by factors such as dense occlusion is better than that of the traditional SSD. |
| first_indexed | 2024-03-11T06:57:05Z |
| format | Article |
| id | doaj.art-07eda3d3d82e405bad35e2e5675afb60 |
| institution | Directory Open Access Journal |
| issn | 2076-3417 |
| language | English |
| last_indexed | 2024-03-11T06:57:05Z |
| publishDate | 2023-03-01 |
| publisher | MDPI AG |
| record_format | Article |
| series | Applied Sciences |
| spelling | doaj.art-07eda3d3d82e405bad35e2e5675afb602023-11-17T09:30:34ZengMDPI AGApplied Sciences2076-34172023-03-01136403810.3390/app13064038ADSSD: Improved Single-Shot Detector with Attention Mechanism and Dilated ConvolutionJian Ni0Rui Wang1Jing Tang2School of Information and Electrical Engineering, Hebei University of Engineering, Handan 056038, ChinaSchool of Information and Electrical Engineering, Hebei University of Engineering, Handan 056038, ChinaSchool of Information and Electrical Engineering, Hebei University of Engineering, Handan 056038, ChinaThe detection of small objects is easily affected by background information, and a lack of context information makes detection difficult. Therefore, small object detection has become an extremely challenging task. Based on the above problems, we proposed a Single-Shot MultiBox Detector with an attention mechanism and dilated convolution (ADSSD). In the attention module, we strengthened the connection between information in space and channels while using cross-layer connections to accelerate training. In the multi-branch dilated convolution module, we combined three expansion convolutions with different dilated ratios to obtain multi-scale context information and used hierarchical feature fusion to reduce the gridding effect. The results show that on PASCAL VOC2007 and VOC2012 datasets, our 300 × 300 input ADSSD model reaches 78.4% mAP and 76.1% mAP. The results outperform those of SSD and other advanced detectors; the effect of some small object detection is significantly improved. Moreover, the performance of the ADSSD in object detection affected by factors such as dense occlusion is better than that of the traditional SSD.https://www.mdpi.com/2076-3417/13/6/4038SSDsmall object detectionattention mechanismmulti-branch dilated convolution |
| spellingShingle | Jian Ni Rui Wang Jing Tang ADSSD: Improved Single-Shot Detector with Attention Mechanism and Dilated Convolution Applied Sciences SSD small object detection attention mechanism multi-branch dilated convolution |
| title | ADSSD: Improved Single-Shot Detector with Attention Mechanism and Dilated Convolution |
| title_full | ADSSD: Improved Single-Shot Detector with Attention Mechanism and Dilated Convolution |
| title_fullStr | ADSSD: Improved Single-Shot Detector with Attention Mechanism and Dilated Convolution |
| title_full_unstemmed | ADSSD: Improved Single-Shot Detector with Attention Mechanism and Dilated Convolution |
| title_short | ADSSD: Improved Single-Shot Detector with Attention Mechanism and Dilated Convolution |
| title_sort | adssd improved single shot detector with attention mechanism and dilated convolution |
| topic | SSD small object detection attention mechanism multi-branch dilated convolution |
| url | https://www.mdpi.com/2076-3417/13/6/4038 |
| work_keys_str_mv | AT jianni adssdimprovedsingleshotdetectorwithattentionmechanismanddilatedconvolution AT ruiwang adssdimprovedsingleshotdetectorwithattentionmechanismanddilatedconvolution AT jingtang adssdimprovedsingleshotdetectorwithattentionmechanismanddilatedconvolution |