Merge pull request #1 from suren3141/master

Video_marker

Author: gihanjayatilaka

Test_load.py

@@ -0,0 +1,5 @@
+import numpy as np
+
+a = np.load("points.npy")
+
+print(a)

VideoMarker.py

@@ -0,0 +1,71 @@
+'''
+USAGE
+python VideoMarker.py FolderWithImages OutputFileName
+
+NOTE : LEFT CLICK for one end and RIGHT CLICK for the other
+Keep pressing ENTER to continue. If you want to stop at the middle press 'q'
+
+LOADING
+numpy.load(OutputFileName.npy)
+
+'''
+
+
+import argparse
+import sys
+import os
+import cv2
+import numpy as np
+
+
+def mouse_callback(event, x, y, flags, param):
+    global frame
+    if event == cv2.EVENT_RBUTTONDOWN:
+        points[2], points[3] = x, y
+        frame1 = np.array(frame, copy=True)
+        cv2.rectangle(frame1,(points[0],points[1]),(points[2],points[3]),(0,255,0),3)
+        cv2.imshow('frame', frame1)
+        # print(x, y, points)
+    if event == cv2.EVENT_LBUTTONDOWN:
+        points[0], points[1] = x, y
+        frame1 = np.array(frame, copy=True)
+        cv2.rectangle(frame1,(points[0],points[1]),(points[2],points[3]),(0,255,0),3)
+        cv2.imshow('frame', frame1)
+        # print(x, y, points)
+
+
+
+if __name__ == "__main__":
+    FILE_NAME=sys.argv[1]
+
+    cv2.namedWindow('frame')
+    cv2.setMouseCallback('frame', mouse_callback)
+
+    cap = cv2.VideoCapture(FILE_NAME)
+    ans = []
+    points = [0, 0, 0, 0]
+
+    while (1):
+        ret, frame = cap.read()
+        if ret == True:
+
+            frame1 = np.array(frame, copy = True)
+            cv2.rectangle(frame1, (points[0], points[1]), (points[2], points[3]), (0, 255, 0), 2)
+            cv2.imshow('frame', frame1)
+
+            k = cv2.waitKey(0) & 0xff
+            if k == ord('q'):
+                break
+
+            print (points)
+            x1, y1, x2, y2 = points
+            points[0] = min(x1, x2)
+            points[1] = min(y1, y2)
+            points[2] = max(x1, x2)
+            points[3] = max(y1, y2)
+
+            ans.append(points)
+
+
+    print("Saved")
+    np.save(sys.argv[2], ans)