2011-07-12 14:56:03 +02:00
|
|
|
import cv2.cv as cv
|
2011-05-17 17:52:23 +02:00
|
|
|
|
|
|
|
def precornerdetect(image):
|
2012-10-17 09:12:04 +02:00
|
|
|
# assume that the image is floating-point
|
2011-05-17 17:52:23 +02:00
|
|
|
corners = cv.CloneMat(image)
|
|
|
|
cv.PreCornerDetect(image, corners, 3)
|
|
|
|
|
|
|
|
dilated_corners = cv.CloneMat(image)
|
|
|
|
cv.Dilate(corners, dilated_corners, None, 1)
|
|
|
|
|
|
|
|
corner_mask = cv.CreateMat(image.rows, image.cols, cv.CV_8UC1)
|
|
|
|
cv.Sub(corners, dilated_corners, corners)
|
|
|
|
cv.CmpS(corners, 0, corner_mask, cv.CV_CMP_GE)
|
|
|
|
return (corners, corner_mask)
|