# -*- coding: utf-8 -*-

import cv2
import numpy as np

filename = '../data/chess.jpg'
img = cv2.imread(filename)
original_img = img.copy()
gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)

gray = np.float32(gray)
dst = cv2.cornerHarris(gray, 2, 5, 0.04)

#result is dilated for marking the corners, not important
dst = cv2.dilate(dst, None)

# Threshold for an optimal value, it may vary depending on the image.
img[dst > (0.01 * dst.max())] = [0, 0, 255]

cv2.imshow('dst', img)
cv2.imshow('original', original_img)

if cv2.waitKey(0) & 0xff == 27:
    cv2.destroyAllWindows()
    
    