The present application belongs to the field of medical
image segmentation of
computer vision, and particularly relates to a
retinal blood vessel segmentation method based on UNet and
Transformer fusion, which has the following characteristics: step 1, pre-
processing the image to be trained to obtain a pre-processed image; step 2, inputting the pre-processed image into a
retinal blood vessel segmentation model based on UNet and
Transformer fusion to obtain a weight file, the model comprising an
encoder, a decoder and a fusion attention mechanism, the
encoder comprising a multi-flow cascaded
convolution layer, a plurality of
pooling layers and a plurality of residual modules, each
convolution layer using a residual module, and the
pooling layer being arranged between two
convolution layer units; the decoder comprising a plurality of improved residual modules based on extended convolution, a plurality of up-sampling modules and a
deconvolution layer, the up-sampling module being arranged between two adjacent improved residual modules based on extended convolution; the fusion attention mechanism taking the output of the
pooling layer in the
encoder and the output of the adjacent flow pooling layer as low-level feature input and high-level feature input, respectively, and the high-level feature input of the fusion attention mechanism at the third layer being a feature map formed by the
Transformer module corresponding to the encoder pooling layer; step 3, loading the weight file and inputting a test
fundus image into the model to obtain a
retinal blood vessel segmentation result. In addition, the
retinal blood vessel segmentation model of the present application is more sensitive to small blood vessels, and the segmentation accuracy is improved.