python数组移位_计算两个数组之间偏移量的简单方法？Python

我有两个dicom图像，可以使用下面的代码计算图像的均方误差。然而，与另一幅图像相比，一幅图像可能存在固有的偏移(如果我的成像仪有点错位)。有没有一种简单的方法来计算两个numpy数组的移位？在

我尝试过将数组每个方向移动几个像素，并计算最小MSQ。然而，这还不够强大。任何帮助或建议将不胜感激！在import numpy as np

import dicom

#first image

ds = dicom.read_file("U:\\temp\\1.dcm")

array1 = ds.pixel_array

#second image

ds1 = dicom.read_file("U:\\temp\\5.dcm")

array2 = ds1.pixel_array

#shifting image by a number of pixels in any direction

arr1 = np.roll(array2, 100, axis=1)

imshow(arr1)

def mse(imageA, imageB):

# the 'Mean Squared Error' between the two images is the

# sum of the squared difference between the two images;

err = np.sum((imageA.astype("float") - imageB.astype("float")) ** 2)

err /= float(imageA.shape[0] * imageA.shape[1])

# return the MSE, the lower the error, the more "similar"

# the two images are

return err

first_try = mse(array1, array2)

second_try = mse(arr1, array2)