Init

main.py

@@ -0,0 +1,110 @@
+import numpy as np
+import cv2
+import datetime
+import matplotlib.pyplot as plt
+import matplotlib
+
+matplotlib.use('WebAgg')
+scale = 2
+max_count = 1088*1920*255
+
+def processVideo(path):
+    prev_gray = None
+    data = []
+    cap = cv2.VideoCapture(path)
+    alpha = 0.90
+    posx = 0.0
+    posy = 0.0
+    posxs = []
+    posys = []
+    counts = []
+    while True:
+        ret, src = cap.read()
+        if not ret:
+            print(ret)
+            break
+        else:
+            smol = cv2.resize(
+                src, (int(src.shape[1]/scale), int(src.shape[0]/scale)))
+            gray = cv2.cvtColor(smol, cv2.COLOR_BGR2GRAY)
+
+            if prev_gray is not None:
+                diff = cv2.absdiff(gray, prev_gray)
+                blur = cv2.GaussianBlur(diff, (5,5),0)
+                retval, thresh = cv2.threshold(blur, 50, 255, cv2.THRESH_TOZERO)
+                cv2.imshow("thresh", thresh)
+                count = np.sum(thresh)/max_count * 100.0
+                if count > 0.001 and count < 0.1:
+                    moments = cv2.moments(thresh)
+                    cX = int(moments["m10"] / moments["m00"])
+                    cY = int(moments["m01"] / moments["m00"])
+                    posx = posx * alpha + cX * (1.0-alpha)
+                    posy = posy * alpha + cY * (1.0-alpha)
+                    posxs.append(posx)
+                    posys.append(posy)
+                    counts.append(count)
+                  
+                cv2.putText(smol, "{:.2}".format(count), (20, 50),
+                            cv2.FONT_HERSHEY_COMPLEX, 1, (255, 0, 0), 2)
+                i = 0
+                while i < len(posxs):
+                    blue_color = 0
+                    if counts[i] > 0.01:
+                        blue_color = 255
+                    else:
+                        blue_color = counts[i]/0.01 * 255
+                        
+                    cv2.circle(smol, (int(posxs[i]), int(posys[i])), 10, (blue_color,0,0),6)
+                    i+= 1
+
+                
+
+                if count > 0.1:
+                    count = 0.1
+                data.append(count)
+
+            cv2.imshow("src", smol)
+            prev_gray = np.copy(gray)
+
+        key = cv2.waitKey(10)
+        if key == 27:
+            break
+    cap.release()
+    cv2.destroyAllWindows()
+    return data
+
+
+out = open("data.tsv", "w+")
+data1 = processVideo("data/curl.mp4")
+data2 = processVideo("data/press.mp4")
+data3 = processVideo("data/situp.mp4")
+
+i = 0
+max_length = 0
+if len(data1) > max_length:
+    max_length = len(data1)
+if len(data2) > max_length:
+    max_length = len(data2)
+if len(data3) > max_length:
+    max_length = len(data3)
+
+while i < max_length:
+    d1 = 0
+    try:
+        d1 = data1[i]
+    except:
+        print("")
+    d2 = 0
+    try:
+        d2 = data2[i]
+    except:
+        print("")
+    d3 = 0
+    try:
+        d3 = data3[i]
+    except:
+        print("")
+    out.write("{}\t{}\t{}\n".format(d1, d2, d3))
+    i += 1
+
+out.close()