diff --git a/httomolibgpu/recon/rotation.py b/httomolibgpu/recon/rotation.py index 3c26fb42..b8101c05 100644 --- a/httomolibgpu/recon/rotation.py +++ b/httomolibgpu/recon/rotation.py @@ -26,8 +26,8 @@ cupy_run = False try: import cupy as xp - import cupyx - from cupyx.scipy.ndimage import gaussian_filter, shift + + # from cupyx.scipy.ndimage import gaussian_filter, shift try: xp.cuda.Device(0).compute_capability @@ -39,13 +39,6 @@ except ImportError: import numpy as np -try: - from cucim.skimage.registration import phase_cross_correlation -except ImportError: - print( - "Cucim library of Rapidsai is a required dependency for find_center_pc method, please install" - ) - import nvtx import math from typing import List, Literal, Optional, Tuple @@ -750,6 +743,12 @@ def find_center_pc(proj1, proj2, tol=0.5, rotc_guess=None): float Rotation axis location. """ + if cupy_run: + from cupyx.scipy.ndimage import shift + from cucim.skimage.registration import phase_cross_correlation + else: + from scipy.ndimage import shift + from skimage.registration import phase_cross_correlation imgshift = 0.0 if rotc_guess is None else rotc_guess - (proj1.shape[1] - 1.0) / 2.0