파이썬 numpy / scipy의 제곱 차이 합계 (SSD)

이미지 처리 알고리즘을 구현하기 위해 Python과 Numpy / Scipy를 사용하려고합니다. 프로파일 러는 두 이미지 간의 제곱 차이의 합을 알려주는 다음 함수 (자주 호출 됨)에 많은 시간이 소요되고 있다고 알려줍니다.

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

속도를 높이려면 어떻게해야합니까? 감사.

 

해결 방법

 

다만

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

(내가 예상하는 것은 모양이 항상 (, , 3) 인 경우 sum ((A-B) ** 2) 일 가능성이 높습니다.)

sum 메서드를 사용할 수도 있습니다 : ((A-B) ** 2) .sum ()

권리?

 

참조 페이지 https://stackoverflow.com/questions/2284611