aboutsummaryrefslogtreecommitdiff
path: root/scripts/.local/bin/personal/edgedetect
diff options
context:
space:
mode:
Diffstat (limited to 'scripts/.local/bin/personal/edgedetect')
-rwxr-xr-xscripts/.local/bin/personal/edgedetect31
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()