The invention belongs to the technical field of
infrared target detection, and particularly relates to an
infrared small target detection method based on a
diffusion model, which comprises the following steps: S1, constructing a conditional
diffusion network model of a double-
branch architecture, the model integrates a conditional network
branch for fully extracting prior features from an input
infrared image # imgabs0 # and a denoising network
branch for recovering a clean
mask # imgabs2 # from a
noise mask # imgabs1 #; s2, constructing a two-dimensional Transform module, and remarkably enhancing the recognition capability of the model on the
global structure and detail features of a target under a complex background through
conjoint analysis of space and channel dimensions; and S3, designing a cross attention fusion module, and realizing accurate guidance of the condition prior branch information on the de-noising branch by calculating the cross attention between the condition prior branch and the de-noising branch. The technical problem of improving the accuracy and reliability of infrared
small target recognition is solved. According to the method, an infrared
small target detection task is converted into a generation normal
form solution, and the accuracy and reliability of small target recognition are remarkably improved.