wassname/scikit-image
1
0
Fork 0
mirror of https://github.com/wassname/scikit-image.git synced 2026-09-10 12:35:06 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix corner_subpix function for corners near border

Browse Source
This commit is contained in:
Johannes Schönberger
2013-11-03 09:16:45 +01:00
parent 610f5898af
commit 40fc5d203f
2 changed files with 26 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
skimage/feature/tests/test_corner.py
+17 -1
View File
@@ -1,5 +1,5 @@
import numpy as np
from numpy.testing import assert_array_equal
from numpy.testing import assert_array_equal, assert_almost_equal
from skimage import data
from skimage import img_as_float
@@ -101,6 +101,22 @@ def test_subpix():
assert_array_equal(subpix[0], (24.5, 24.5))
def test_subpix_border():
img = np.zeros((50, 50))
img[1:25,1:25] = 255
img[25:-1,25:-1] = 255
corner = corner_peaks(corner_harris(img), min_distance=1)
subpix = corner_subpix(img, corner, window_size=11)
ref = np.array([[ 0.52040816, 0.52040816],
[ 0.52040816, 24.47959184],
[24.47959184, 0.52040816],
[24.5 , 24.5 ],
[24.52040816, 48.47959184],
[48.47959184, 24.52040816],
[48.47959184, 48.47959184]])
assert_almost_equal(subpix, ref)
def test_num_peaks():
"""For a bunch of different values of num_peaks, check that
peak_local_max returns exactly the right amount of peaks. Test