Air-based rail detection method based on attention enhancement mechanism

Abstract

The invention discloses an air-based rail detection method based on an attention enhancement mechanism, belonging to the field of aviation surveillance. The method comprises the steps that firstly, the high-definition array camera is mounted on the aerial vehicle, and the original image of the railway track is acquired and pre-processed to obtain training samples; the images of the training samples are inputted into the convolutional neural network for feature extraction, and the 16*16 encoded images are obtained, and the attention enhancement is carried out to increase the ability of featureexpression; then the up sampling is carried out on the image after the attention enhancement by convolution neural network to get the same size decision map as the original image, and the decision mapis classified and judged, and a point in the decision map is a point on the rail; and finally, according to the decision result, the network parameters are updated by back propagation using gradientdescent algorithm. On the basis of precise detection, the method of the invention can monitor the surrounding environment of the rail and the safety condition of the rail, has accurate positioning andremarkable effect, and has strong engineering value and practical significance.

Description

technical field [0001] The invention belongs to the field of aviation surveillance, in particular to an air-based rail detection method based on an attention enhancement mechanism. Background technique [0002] China is a vast country with a vast territory of 9.6 million square kilometers. The huge land area also puts forward high requirements for the construction of transportation in our country. In order to meet the needs of national economic construction and people's daily travel, our country has developed a variety of transportation systems. Among them, the railway is the most important and popular, and plays a key role in China's comprehensive transportation system. Its economical and fast characteristics make it a widely used and popular mode of transportation. As of the end of 2016, China's total operating mileage of railways reached 124,000 kilometers, ranking second in the world; among them, 25,000 kilometers of high-speed railways ranked first in the world. con...

AI Technical Summary

Problems solved by technology

However, the flying altitude of the airborne aircraft is relatively high, so the target object in the collected pictures may be relatively small

Images