An improved target detection algorithm based on EfficientNet
Tao Wu, Hongjin Zhu, Honghui Fan, Hongyan Zhou · Journal of Physics Conference Series · 2021
Abstract In order to improve the detection accuracy for small-scale targets in complex scenes, an improved target detection algorithm based on EfficientNet is proposed. Firstly, the EfficientNet network is used to optimize the DarkNet53 feature extraction network. Compressing the standard convolution with the depthwise separable convolution, and increasing the depth of the neural network with the residual network that can effectively achieve feature extraction, reduce the number of parameters and improve the detection speed. Secondly, a feature pyramid network is used to design four scales of features for multi-scale feature extraction, which improves the detection of small targets. The experimental results show that the YOLOv3 target detection algorithm improves the detection accuracy by 4.98% compared to the original algorithm on VOC dataset, which improves the detection accuracy and ensures real-time detection for small targets.