| CPC G06T 7/337 (2017.01) [G06T 3/4007 (2013.01); G06T 7/32 (2017.01); G06T 2207/20081 (2013.01)] | 3 Claims |

|
1. A method for training a non-rigid registration model for images, comprising:
(a) inputting a preprocessed reference image and a preprocessed floating image to a U-Net to obtain spatial transformation parameters, wherein the preprocessed reference image is obtained using rigid registration based on an iterative closest point registration and a mutual information registration, wherein rigid registration includes performing coarse registration about contour point data sets on the preprocessed reference image and the preprocessed floating image to be registered by using iterative closest point registration, and a contour point data set of each of the preprocessed reference image and the preprocessed floating image is extracted through a marching cubes algorithm;
(b) inputting the spatial transformation parameters to a spatial transformation network, and performing spatial transformation and an interpolation operation on the preprocessed floating image, so as to obtain a registration result image;
(c) calculating a loss function value between the preprocessed reference image and the registration result image by using a loss function, wherein the loss function comprises both a correlation coefficient and a mean squared error between the preprocessed reference image and the registration result image; and
(d) repeating steps (a) to (c) a predetermined number of times for iterative training or until the non-rigid registration model converges.
|