translation.py 文件源码

def initSize(self,size):
        w,h = size
        w = int(round(w))
        h = int(round(h))

        if self.x.has_key((w,h)):
            return

        # Initializing translations

        self.x[(w,h)] = []
        for x in range(w):
            mat = np.zeros((w,h),dtype=np.complex64)
            mat[x,0] = 1.0
            filter = np.fft.fft2(mat)
            self.x[(w,h)].append(filter)

        self.y[(w,h)] = []
        for y in range(h):
            mat = np.zeros((w,h),dtype=np.complex64)
            mat[0,y] = 1.0
            filter = np.fft.fft2(mat)
            self.y[(w,h)].append(filter)