debuq

predict_pose.py

@@ -532,4 +532,34 @@ def predict_pose(video: List[np.ndarray], models: tuple, sign_space=4, yolo_sign
         results["sign_space"].append(pad_bbox)
     results["images"] = video
 
+    # ==========================================
+    # NOVÉ: ULOŽENÍ KEYPOINTS DO JSON SOUBORU
+    # ==========================================
+    
+    # 1. Vytvoříme složku, kam se budou JSONy ukládat
+    output_dir = "pose_results"
+    os.makedirs(output_dir, exist_ok=True)
+    
+    # 2. Vytvoříme unikátní jméno souboru s časovým razítkem
+    timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
+    json_filename = os.path.join(output_dir, f"keypoints_{timestamp}.json")
+    
+    # 3. Připravíme data pro JSON.
+    # Musíme je vybrat z results, protože ty obsahují i obrázky (numpy arrays), 
+    # které do JSONu nejdou přímo uložit.
+    json_data = {
+        "keypoints": results["keypoints"],
+        "cropped_keypoints": results["cropped_keypoints"]
+    }
+    
+    # 4. Uložíme do souboru
+    try:
+        with open(json_filename, "w", encoding="utf-8") as f:
+            json.dump(json_data, f, indent=4)
+        print(f"DEBUG: Keypoints úspěšně uloženy do souboru: {json_filename}")
+    except Exception as e:
+        print(f"VAROVÁNÍ: Nepodařilo se uložit JSON soubor. Chyba: {e}")
+        
+    # ==========================================
+
     return results