working

moves_dump.py

@@ -0,0 +1,42 @@
+import json
+import pickle
+import time
+
+import cv2
+import numpy as np
+
+import utils
+from draw import draw_new
+
+moves = []
+better_moves = {}
+
+with open('moves.pkl', 'rb') as f:  # 'rb' = read binary
+    moves = pickle.load(f)
+
+startValue = moves[0][0]
+totalCount = len(moves)
+
+for i, move in enumerate(moves):
+    moves[i] = (move[0] - startValue, move[1], move[2])
+
+    # left_hip = move[2][11]   # Left Hip
+    # right_hip = move[2][12]  # Right Hip
+    # center = (left_hip + right_hip) / 2
+    #
+    # # Normalizacja względem środka ciała
+    # normalized_keypoints = move[2] - center
+    #
+    # better_moves[round((move[0] - startValue) * 1000)] = normalized_keypoints.tolist()
+    #
+    # # scale = utils.distance(move[2][11], move[2][12])
+    # # print(scale)
+    # draw = normalized_keypoints + 200
+
+
+    # Do rysowania (np. przesunięcie na ekran)
+    draw = utils.normalize(move[2])
+
+    cv2.imshow('you', draw_new(draw))
+    cv2.waitKey(1)
+    time.sleep(0.1)