Merge pull request #34 from andreped/demo-fix

DeepDeformationMapRegistration/main.py

@@ -310,6 +310,7 @@ def main():
             # disp_map = disp_map_or
             pred_image = zoom(pred_image, 1 / zoom_factors)
             pred_image = pad_crop_to_original_shape(pred_image, fixed_image_or.shape, crop_min)
+            pred_image = np.squeeze(pred_image, axis=-1)
             LOGGER.info('Done...')
 
         if args.original_resolution:

demo/src/gui.py

@@ -61,7 +61,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 = load_ct_to_numpy(output_path + "pred_image.nii.gz")
-        return self.pred_images
+        return None
 
     def get_fixed_image(self, k):
         k = int(k) - 1
@@ -183,7 +183,7 @@ class WebUI:
                             self.run_btn.click(
                                 fn=lambda x: self.process(x),
                                 inputs=file_output,
-                                outputs=t,
+                                outputs=None,
                             )
 
                         self.slider.input(

demo/src/utils.py

@@ -21,7 +21,6 @@ def load_ct_to_numpy(data_path):
     data = data / np.amax(data) * 255
     data = data.astype("uint8")
 
-    print(data.shape)
     return [data[..., i] for i in range(data.shape[-1])]
 
 
@@ -38,7 +37,6 @@ def load_pred_volume_to_numpy(data_path):
     data[data > 0] = 1
     data = data.astype("uint8")
 
-    print(data.shape)
     return [data[..., i] for i in range(data.shape[-1])]