#!/usr/bin/env python
# -*- coding: utf-8 -*-

import numpy as np
import cv2

src = cv2.imread('fig9-9a.jpg')
cv2.imshow("src",src)

# 角点响应函数R
gray = cv2.cvtColor(src,cv2.COLOR_BGR2GRAY)
gray = np.float32(gray)
R = cv2.cornerHarris(gray,2,3,0.05)
cv2.imshow("R",R)

# 膨胀，以标注角点
dst = cv2.dilate(R,None)
# 提取角点，具体阈值系数与图像有关
src[dst>0.03*dst.max()]=[0,0,255]
harris = src.copy()
cv2.imshow('dst',harris)


cv2.imwrite("fig9-9b.png",R)
cv2.imwrite("fig9-9c.png",harris)

cv2.waitKey(0)
cv2.destroyAllWindows()