mirror of
https://github.com/wassname/scikit-image.git
synced 2026-06-28 04:39:13 +08:00
Johannes Schönberger
39 lines
1.1 KiB
Python
39 lines
1.1 KiB
Python
"""
|
|
================
|
|
Corner detection
|
|
================
|
|
|
|
Detect corner points using the Harris corner detector and determine subpixel
|
|
position of corners.
|
|
|
|
.. [1] http://en.wikipedia.org/wiki/Corner_detection
|
|
.. [2] http://en.wikipedia.org/wiki/Interest_point_detection
|
|
|
|
"""
|
|
|
|
import numpy as np
|
|
from matplotlib import pyplot as plt
|
|
|
|
from skimage import data
|
|
from skimage.feature import corner_harris, corner_subpix, peak_local_max
|
|
from skimage.transform import warp, AffineTransform
|
|
from skimage.draw import ellipse
|
|
|
|
tform = AffineTransform(scale=(1.3, 1.1), rotation=1, shear=0.7,
|
|
translation=(210, 50))
|
|
image = warp(data.checkerboard(), tform.inverse, output_shape=(350, 350))
|
|
rr, cc = ellipse(310, 175, 10, 100)
|
|
image[rr, cc] = 1
|
|
image[180:230, 10:60] = 1
|
|
image[230:280, 60:110] = 1
|
|
|
|
coords = peak_local_max(corner_harris(image), min_distance=5)
|
|
coords_subpix = corner_subpix(image, coords, window_size=13)
|
|
|
|
plt.gray()
|
|
plt.imshow(image, interpolation='nearest')
|
|
plt.plot(coords[:, 1], coords[:, 0], '.b', markersize=3)
|
|
plt.plot(coords_subpix[:, 1], coords_subpix[:, 0], '+r', markersize=15)
|
|
plt.axis((0, 350, 350, 0))
|
|
plt.show()
|