Created lines for detection

Author: sampreets3

scripts/__pycache__/utils.cpython-36.pyc

@@ -49,9 +49,14 @@
 line_end   = (800, 500)
 
 line_color = (255, 0, 0)
+alt_line_color = (0, 0, 255)
 line_thickness = 2
 #---------------------------------------------------------
 
+# Global variable to count trees -------------------------
+treeNum = 0
+#---------------------------------------------------------
+
 # Load the network----------------------------------------
 #net = cv2.dnn.readNetFromDarknet(config, weights)
 net = cv2.dnn.readNetFromDarknet(config, weights)
@@ -114,8 +119,13 @@
                                 c_x, c_y = c[0], c[1]
                                 cv2.circle(img, (c_x, c_y), radius=2, color=(0, 255, 255), thickness=2)
                                 cv2.line(img, line_start, line_end, line_color, line_thickness)
-                                cv2.putText(img, str(i), (100, 100), cv2.FONT_HERSHEY_SIMPLEX, 1, (128, 120, 68), 2)
 
+                                #if(c_y > 500):
+                                #    treeNum = treeNum + 1
+
+                                #cv2.putText(img, str(treeNum), (100, 100), cv2.FONT_HERSHEY_SIMPLEX, 1, (128, 120, 68), 2)
+
+                                cv2.line(img, (c_x, c_y), (c_x, 500), alt_line_color, line_thickness)
         # Display the resulting frame
         cv2.imshow('image',img)
         out.write(img)

scripts/detect-trees.py

@@ -1,2 +1,13 @@
 def get_centroid(x1, y1, x2, y2):
     return int((x1+x2)/2) ,int((y1 + y2)/2)
+
+def is_lower_than_line(l_start_x, l_start_y, l_end_x, l_end_y, pt_x, pt_y):
+    d = ((pt_x - l_start_x)*(l_end_y - l_start_y)) - ((pt_y - l_start_y)*(l_end_x - l_start_x))
+    if d < 0:
+        return 0
+
+    elif d == 0:
+        return 1
+
+    else :
+        return 2