diff options
Diffstat (limited to 'scripts/.local/bin/personal/edgedetect')
-rwxr-xr-x | scripts/.local/bin/personal/edgedetect | 31 |
1 files changed, 31 insertions, 0 deletions
diff --git a/scripts/.local/bin/personal/edgedetect b/scripts/.local/bin/personal/edgedetect new file mode 100755 index 0000000..34d26c8 --- /dev/null +++ b/scripts/.local/bin/personal/edgedetect @@ -0,0 +1,31 @@ +#!/bin/env python3 + +import sys +import cv2 as cv +import numpy as np +import matplotlib.pyplot as plt +from skimage.io import imread, imshow +from skimage.color import rgb2gray +from skimage.feature import match_template, peak_local_max +from skimage import transform + +def detect(): + #img = cv.imread(cv.samples.findFile(sys.argv[1])) + #if img is None: + # sys.exit("Could not read the image.") + #_,img = cv.threshold(img, 128, 255, cv.THRESH_BINARY) + + image= cv.imread(sys.argv[1]) + gray= cv.cvtColor(image, cv.COLOR_BGR2GRAY) + gray= np.float32(gray) + harris_corners= cv.cornerHarris(gray, 3, 3, 0.05) + kernel= np.ones((7,7), np.uint8) + harris_corners= cv.dilate(harris_corners, kernel, iterations= 2) + image[harris_corners > 0.025 * harris_corners.max()]= [255,127,127] + cv.imshow('Harris Corners', image) + cv.waitKey(0) + cv.destroyAllWindows() + + +if __name__ == "__main__": + detect() |