The invention provides a three-dimensional MRI (
Magnetic Resonance Imaging)
brain tumor segmentation method based on a
diffusion model, and relates to the technical field of medical
image generation, and the method comprises the steps: collecting MRI images of different
modes of a
brain tumor part and corresponding segmentation reference true values; performing
noise addition on the segmentation reference true value through forward
diffusion to obtain
noise-added segmentation data; performing iterative
discrete wavelet transform on the
MRI image to obtain
wavelet information of each level; constructing a
brain tumor segmentation model for improving a
diffusion model on the basis of introducing a cross attention mechanism into a U-Net
network structure, and training the brain tumor segmentation model by using the data; a
random noise is sampled from prior distribution, the
random noise and a to-be-segmented
MRI image are spliced and then input into the trained model, a target domain segmentation image corresponding to reasoning is obtained, the method can extract multi-scale features of multi-
modal MRI, the parameter quantity of the model is remarkably reduced, condition information is fused into a network, feature expression of a key area is enhanced, and the target domain segmentation image is obtained. Therefore, the accuracy of
MRI brain tumor segmentation is improved.