In order to solve the problems of large number of deep model parameters and poor real-time performance in the existing vehicle detection methods, this paper proposes a vehicle detection method based on Taylor expansion model pruning algorithm. VGG16 is the main trunk network, the Taylor expansion method is is used for filter pruning, and the Lagrange algorithm is combined. Experiments show that the vehicle detection method proposed in this paper can reduce the complexity of the model, reduce the number of model parameters by 90%, and shorten the detection time without affecting the detection accuracy, which has certain application and promotion value.
|