improve test coverage. Use SciPy's fourier_shift for examples.

This commit is contained in:
Michael Sarahan
2015-02-21 11:33:37 -08:00
parent cf09b0c5ef
commit 71a89697f0
3 changed files with 65 additions and 62 deletions
+4 -2
View File
@@ -20,12 +20,14 @@ import matplotlib.pyplot as plt
from skimage import data
from skimage.feature import register_translation
from skimage.feature.register_translation import _upsampled_dft, fourier_shift
from skimage.feature.register_translation import _upsampled_dft
from scipy.ndimage.fourier import fourier_shift
image = data.camera()
shift = (-2.4, 1.32)
# (-2.4, 1.32) pixel offset relative to reference coin
offset_image = fourier_shift(image, shift)
offset_image = fourier_shift(np.fft.fftn(image), shift)
offset_image = np.fft.ifftn(offset_image)
print("Known offset (y, x):")
print(shift)