edge_detection algorithms added

Author: ratan

Image_Processing/src/edge_detection/harris_corner_detection/harris_corner_detection.py

@@ -0,0 +1,30 @@
+# Import necessary libararies
+from matplotlib import pyplot as plt
+from skimage.io import imread
+from skimage.color import rgb2gray
+from skimage.feature import corner_harris, corner_subpix, corner_peaks
+
+
+def main(inputfilename, oututfilename):
+    # Load image
+    image = imread(inputfilename)
+    image = rgb2gray(image)
+
+    # Apply corner detection algorithm
+    corners = corner_harris(image)
+    coords = corner_peaks(corners, min_distance=5)
+    coords_subpix = corner_subpix(image, coords, window_size=13)
+
+    # Diplay the image
+    fig, ax = plt.subplots()
+    ax.imshow(image, interpolation='nearest', cmap=plt.cm.gray)
+    ax.plot(coords[:, 1], coords[:, 0], '.b', markersize=3)
+    ax.plot(coords_subpix[:, 1], coords_subpix[:, 0], '+r', markersize=15)
+    ax.axis((0, 350, 350, 0))
+    plt.savefig(oututfilename)
+
+
+if __name__ == "__main__":
+    inputfilename = input("Enter a input filename:\t")
+    oututfilename = input("Enter a output filename:\t")
+    main(inputfilename, oututfilename)

Image_Processing/src/edge_detection/hough_line_edge_detector/hough_line_edge_detector.py

@@ -0,0 +1,26 @@
+import cv2
+import numpy as np
+
+
+def main(inputfilename, oututfilename):
+    img = cv2.imread(inputfilename)
+    gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
+
+    edges = cv2.Canny(gray, 100, 200, apertureSize=3)
+    cv2.imshow('edges', edges)
+    cv2.waitKey(0)
+
+    minLineLength = 30
+    maxLineGap = 10
+    lines = cv2.HoughLinesP(edges, 1, np.pi/180, 15, minLineLength, maxLineGap)
+    for x in range(0, len(lines)):
+        for x1, y1, x2, y2 in lines[x]:
+            cv2.line(img, (x1, y1), (x2, y2), (0, 255, 0), 2)
+
+    cv2.imwrite(oututfilename, img)
+
+
+if __name__ == "__main__":
+    inputfilename = input("Enter a input filename:\t")
+    oututfilename = input("Enter a output filename:\t")
+    main(inputfilename, oututfilename)

Image_Processing/src/edge_detection/sobel_edge_detection/sobel_edge_detection.py

@@ -0,0 +1,24 @@
+# Import neccessary libraries
+from skimage import io, filters, color
+
+
+def main(inputfilename, oututfilename):
+
+    # Read image
+    img = io.imread(inputfilename)
+
+    # Convert RGB to gray
+    img = color.rgb2gray(img)
+
+    # For edge detection
+    edge = filters.sobel(img)
+
+    # Display and save image
+    io.imshow(edge)
+    io.imsave(oututfilename, edge)
+
+
+if __name__ == "__main__":
+    inputfilename = input("Enter a input filename:\t")
+    oututfilename = input("Enter a output filename:\t")
+    io = main(inputfilename, oututfilename)