Return
Multi-object garbage image detection algorithm based on SP-SSD
DOI:10.1016/j.eswa.2024.125773.png)
Abstract
En 中文
The existing garbage image detection methods are large in scale, long detection response time and low in accuracy. To solve these problems, this paper propose a multi-objective garbage image detection algorithm based on Self-Improvement Single Shot MultiBox Detector (SP-SSD). In the stage of feature extraction, depthwise separable convolution is used for downsampling, and a expansion convolution module (ECM) is designed to extract features. In the feature information fusion stage, residual connections are introduced for enhancing the generalization ability. In the stage of feature weight allocation, the spatial pyramid attention module (SPA) is introduced to enhance the representation ability. To verify the performance of the algorithm, this paper used 10,000 garbage images from the garbage dataset GCDD to carry out experiments. Experiments showed that compared with the classical SSD model, the mAP of SP-SSD is increased by 14.52%, parameters are reduced by 7 times, and the FPS is increased by 7.
Keywords:
Garbage detection
ECM convolution module
Residual connection
Spatial pyramid attention SPA module
Journal
IF:
7.5
Papers:
2.9W
Citations:
10.2W
Organization
No organization information available

