Vehicle detection method based on convolutional neural network
暂无分享,去创建一个
The invention discloses a vehicle detection method based on a convolutional neural network. The method includes the step S1 of collecting vehicle samples and non-vehicle samples and classifying the vehicle samples, the step S2 of preprocessing the samples, the step S3 of training a CNN vehicle detector, the step S4 calculating an average similarity table of a characteristic pattern, the step S5 of constructing a similarity characteristic pattern set, the step S6 of obtaining a CNN-OP vehicle detector, the step S7 of obtaining detection images, the step S8 of preprocessing the obtained detection images, the step S9 of constructing an image pyramid for the detection images, the step S10 of extracting characteristics, the step S11 of scanning characteristic patterns, the step S12 of classifying the characteristics, and the step S13 of combining detection windows and conducting output. An offline optimization scheme is put forward, the convolutional neural network which is completely trained is optimized, the strategy of scanning the windows after extracting the characteristics is adopted at the detection stage, and therefore the characteristics are prevented from being repeatedly calculated, and the detection speed of the system is increased.