Descargo de responsabilidad: esta pregunta es sobre OpenPose, pero la clave aquí es averiguar cómo usar la salida (coordenadas almacenadas en el JSON) y no cómo usar OpenPose, así que considere leerlo hasta el final.
Tengo un video de una persona de lado en una bicicleta (perfil de él sentado para que veamos el lado derecho). Uso OpenPose para extraer las coordenadas del esqueleto. OpenPose proporciona las coordenadas en un archivo JSON que se ve así (consulte los documentos para obtener una explicación ):
{ "version": 1.3, "people": [ { "person_id": [ -1 ], "pose_keypoints_2d": [ 594.071, 214.017, 0.917187, 523.639, 216.025, 0.797579, 519.661, 212.063, 0.856948, 539.251, 294.394, 0.873084, 619.546, 304.215, 0.897219, 531.424, 221.854, 0.694434, 550.986, 310.036, 0.787151, 625.477, 339.436, 0.845077, 423.656, 319.878, 0.660646, 404.111, 321.807, 0.650697, 484.434, 437.41, 0.85125, 404.13, 556.854, 0.791542, 443.261, 319.801, 0.601241, 541.241, 370.793, 0.921286, 502.02, 494.141, 0.799306, 592.138, 198.429, 0.943879, 0, 0, 0, 562.742, 182.698, 0.914112, 0, 0, 0, 537.25, 504.024, 0.530087, 535.323, 500.073, 0.526998, 486.351, 500.042, 0.615485, 449.168, 594.093, 0.700363, 431.482, 594.156, 0.693443, 386.46, 560.803, 0.803862 ], "face_keypoints_2d": [], "hand_left_keypoints_2d": [], "hand_right_keypoints_2d": [], "pose_keypoints_3d": [], "face_keypoints_3d": [], "hand_left_keypoints_3d": [], "hand_right_keypoints_3d": [] } ] }Por lo que entiendo, cada JSON es un cuadro del video.
Mi objetivo es detectar los ángulos de coordenadas específicas como la rodilla derecha, el brazo derecho, etc. Por ejemplo:
openpose_angles = [(9, 10, 11, "right_knee"), (2, 3, 4, "right_arm")]Esto se basa en el siguiente maniquí de esqueleto de OpenPose:
Lo que hice fue calcular el ángulo entre tres coordenadas (usando Python):
temp_df = json.load(open(os.path.join(jsons_dir, file))) listPoints = list(zip(*[iter(temp_df['people'][person_number]['pose_keypoints_2d'])] * 3)) count = 0 lmList2 = {} for x,y,c in listPoints: lmList2[count]=(x,y,c) count+=1 p1=angle_cords[0] p2=angle_cords[1] p3=angle_cords[2] x1, y1 ,c1= lmList2[p1] x2, y2, c2 = lmList2[p2] x3, y3, c3 = lmList2[p3] # Calculate the angle angle = math.degrees(math.atan2(y3 - y2, x3 - x2) - math.atan2(y1 - y2, x1 - x2)) if angle < 0: angle += 360 Este método lo vi en algún blog (que olvidé dónde), pero estaba relacionado con OpenCV en lugar de OpenPose (no estoy seguro si hace la diferencia), pero veo ángulos que no tienen sentido. Se lo mostramos a nuestro maestro y nos sugirió que usáramos vectores para calcular los ángulos, en lugar de usar math.atan2 . Pero nos confundimos sobre cómo implementar esto.
Para resumir, aquí está la pregunta: ¿Cuál será la mejor manera de calcular los ángulos? ¿Cómo calcularlos usando vectores?