python - numpy/scipy 中的平方差和 (SSD)

Question

我正在尝试使用 Python 和 Numpy/Scipy 来实现图像处理算法。探查器告诉我在以下函数(经常调用)上花费了大量时间，它告诉我两幅图像之间的平方差之和

def ssd(A,B):
    s = 0
    for i in range(3):
        s += sum(pow(A[:,:,i] - B[:,:,i],2))
    return s

我怎样才能加快速度？谢谢。

Answer 1

只是

s = numpy.sum((A[:,:,0:3]-B[:,:,0:3])**2)

(如果形状始终为 (,,3)，我预计可能只是 sum((A-B)**2))

也可以使用求和方法:((A-B)**2).sum()

对吧？