First upload

arrow_detection_hough_lines.py

@@ -0,0 +1,54 @@
+import cv2, numpy as np, argparse
+
+ap = argparse.ArgumentParser()
+ap.add_argument("-i", "--image", help = "path to the image file")
+args = vars(ap.parse_args())
+img = cv2.imread(args["image"])
+#convert the image to grayscale
+gray = cv2.cvtColor(img,cv2.COLOR_BGR2GRAY)
+#apply canny edge detection to the image
+edges = cv2.Canny(gray,50,150,apertureSize = 3)
+#show what the image looks like after the application of previous functions
+cv2.imshow("canny'd image", edges)
+cv2.waitKey(0)
+#perform HoughLines on the image
+lines = cv2.HoughLines(edges,1,np.pi/180,20)
+#create an array for each direction, where array[0] indicates one of the lines and array[1] indicates the other, which if both > 0 will tell us the orientation
+left = [0, 0]
+right = [0, 0]
+up = [0, 0]
+down = [0, 0]
+#iterate through the lines that the houghlines function returned
+for object in lines:
+    theta = object[0][1]
+    rho = object[0][0]
+    #cases for right/left arrows
+    if ((np.round(theta, 2)) >= 1.0 and (np.round(theta, 2)) <= 1.1) or ((np.round(theta,2)) >= 2.0 and (np.round(theta,2)) <= 2.1):
+        if (rho >= 20 and rho <=  30):
+            left[0] += 1
+        elif (rho >= 60 and rho <= 65):
+            left[1] +=1
+        elif (rho >= -73 and rho <= -57):
+            right[0] +=1
+        elif (rho >=148 and rho <= 176):
+            right[1] +=1
+    #cases for up/down arrows
+    elif ((np.round(theta, 2)) >= 0.4 and (np.round(theta,2)) <= 0.6) or ((np.round(theta, 2)) >= 2.6 and (np.round(theta,2))<= 2.7):
+        if (rho >= -63 and rho <= -15):
+            up[0] += 1
+        elif (rho >= 67 and rho <= 74):
+            down[1] += 1
+            up[1] += 1
+        elif (rho >= 160 and rho <= 171):
+            down[0] += 1
+
+if left[0] >= 1 and left[1] >= 1:
+    print("left")
+elif right[0] >= 1 and right[1] >= 1:
+    print("right")
+elif up[0] >= 1 and up[1] >= 1:
+    print("up")
+elif down[0] >= 1 and down[1] >= 1:
+    print("down")
+
+print(up, down, left, right)