def make_image():
image = np.random.rand(20, 70, 3)
window = cv2.createHanningWindow((image.shape[1], image.shape[0]), cv2.CV_64FC1)
grayscale = np.mean(image, 2)
hanned = grayscale * window
image_fft = PhaseCorrelate.preprocessForPhaseCorrelate(hanned)
return image, image_fft
def preprocess(self, im):
if self._window is None:
self._window = cv2.createHanningWindow((im.shape[1], im.shape[0]), cv2.CV_64FC1)
# grayscale = np.mean(im, 2)
hanned = im * self._window
image_fft = PhaseCorrelate.preprocessForPhaseCorrelate(hanned)
return image_fft