First commit

hpe_ffmpeg.py

@@ -0,0 +1,66 @@
+import sys
+import cv2
+import time
+from hpe_mp_class import hpe_mp_class
+from pythonosc import udp_client
+
+# Arguments
+adress_input = "in.pipe1"
+if len(sys.argv) > 1:
+    adress_input = sys.argv[1]
+    for i in range(0, 100):
+        if adress_input == str(i):
+            adress_input = i
+
+osc_address = "10.199.0.14"
+osc_port = 5005
+client = udp_client.SimpleUDPClient(osc_address, osc_port)
+
+showing = False
+
+# Videocapture
+cap = cv2.VideoCapture(adress_input)
+
+# Preprocessing parameters
+crop = 1.0
+frame_width = int(crop*cap.get(cv2.CAP_PROP_FRAME_WIDTH))
+frame_height = int(crop*cap.get(cv2.CAP_PROP_FRAME_HEIGHT))
+
+# FPS variables
+pTime = 0
+cTime = 0
+
+# Mediapie class
+mp_cl = hpe_mp_class()
+
+while True:
+    # Reading frame
+    success, img = cap.read()
+
+    # Image preprocessing
+    img = cv2.resize(img, (frame_width, frame_height))
+
+    # Mediapipe
+    mp_cl.process(img)
+    mp_cl.show(img)
+
+    if showing:
+        # FPS
+        cTime = time.time()
+        fps = 1. / (cTime - pTime)
+        pTime = cTime
+
+        # Showing
+        img = cv2.flip(img, 1)  # mirror
+        cv2.putText(img, str(int(fps)), (22, 32), cv2.FONT_HERSHEY_COMPLEX, 1, (0, 0, 0), 2)
+        cv2.putText(img, str(int(fps)), (20, 30), cv2.FONT_HERSHEY_COMPLEX, 1, (255, 255, 255), 2)
+        cv2.imshow("Main", img)
+
+    # get converted map
+    res = mp_cl.getJSON()
+    client.send_message("/pose/0", res)
+
+    # Interface
+    key = cv2.waitKey(1)
+    if key == 27:
+        break