Chimney detection method based on ai technology

A chimney detection method based on AI technology, comprising:

• collecting a remote sensing chimney image dataset;
• dividing the remote sensing chimney image dataset into a training set and a validation set and performing data augmentation;
• inputting the remote sensing chimney image dataset into a backbone network for feature extraction and transmitting the remote sensing chimney image dataset to a neck network to extract feature information, which includes spatial information and channel information;
• transmitting feature information obtained from the neck network to a feature pyramid;
• performing up sampling and down sampling for feature fusion;
• strengthening features through a explicit visual center and global attention mechanism to obtain a corresponding enhanced feature map;
• inputting the corresponding enhanced feature map into a head network;
• using four decoupled object detection heads included in the head network to perform object detection to obtain a corresponding predicted feature map;
• and
• outputting a final preselection box through non-maximum suppression to obtain a detection result of the remote sensing chimney image.