Apply spatial transformer to get registered moving image

DeepDeformationMapRegistration/main.py

@@ -297,7 +297,8 @@ def main():
 
         LOGGER.info('Applying displacement map...')
         time_pred_img_start = time.time()
-        pred_image = SpatialTransformer(interp_method='linear', indexing='ij', single_transform=False)([moving_image[np.newaxis, ...], disp_map[np.newaxis, ...]]).eval()
+        pred_image = SpatialTransformer(interp_method='linear', indexing='ij', single_transform=False)([moving_image[np.newaxis, ...], disp_map[np.newaxis, ...]])#.eval()
+        pred_image = np.asarray(pred_image)
         time_pred_img_end = time.time()
         LOGGER.info(f'\t... done ({time_pred_img_end - time_pred_img_start} s)')
         pred_image = pred_image[0, ...]

demo/src/gui.py

@@ -60,7 +60,7 @@ class WebUI:
 
         self.fixed_images = load_ct_to_numpy(fixed_image_path)
         self.moving_images = load_ct_to_numpy(moving_image_path)
-        self.pred_images = np.ones_like(self.moving_images)
+        self.pred_images = load_ct_to_numpy(output_path + "pred_image.nii.gz")
         return self.pred_images
 
     def get_fixed_image(self, k):