Fixed dimension problem in main script
Browse files
DeepDeformationMapRegistration/main.py
CHANGED
|
@@ -310,6 +310,7 @@ def main():
|
|
| 310 |
# disp_map = disp_map_or
|
| 311 |
pred_image = zoom(pred_image, 1 / zoom_factors)
|
| 312 |
pred_image = pad_crop_to_original_shape(pred_image, fixed_image_or.shape, crop_min)
|
|
|
|
| 313 |
LOGGER.info('Done...')
|
| 314 |
|
| 315 |
if args.original_resolution:
|
|
|
|
| 310 |
# disp_map = disp_map_or
|
| 311 |
pred_image = zoom(pred_image, 1 / zoom_factors)
|
| 312 |
pred_image = pad_crop_to_original_shape(pred_image, fixed_image_or.shape, crop_min)
|
| 313 |
+
pred_image = np.squeeze(pred_image, axis=-1)
|
| 314 |
LOGGER.info('Done...')
|
| 315 |
|
| 316 |
if args.original_resolution:
|