| def _create_interaction_row( |
| video_name, frame_data, interaction, frame_width, frame_height |
| ): |
| """Create a row of interaction data for CSV output.""" |
| row = { |
| "video_name": video_name, |
| "frame_index": frame_data["frame_index"], |
| "timestamp": frame_data["timestamp"], |
| "frame_width": frame_width, |
| "frame_height": frame_height, |
| "person1_id": interaction["person1_id"], |
| "person2_id": interaction["person2_id"], |
| "box1_x_min": interaction["box1"][0], |
| "box1_y_min": interaction["box1"][1], |
| "box1_x_max": interaction["box1"][2], |
| "box1_y_max": interaction["box1"][3], |
| "box2_x_min": interaction["box2"][0], |
| "box2_y_min": interaction["box2"][1], |
| "box2_x_max": interaction["box2"][2], |
| "box2_y_max": interaction["box2"][3], |
| "center1_x": interaction["center1"][0], |
| "center1_y": interaction["center1"][1], |
| "center2_x": interaction["center2"][0], |
| "center2_y": interaction["center2"][1], |
| "distance": interaction["distance"], |
| "person1_idx": interaction["person1_idx"], |
| "person2_idx": interaction["person2_idx"], |
| "relative_distance": interaction["relative_distance"], |
| "motion_average_speed": frame_data["motion_features"]["average_speed"], |
| "motion_motion_intensity": frame_data["motion_features"]["motion_intensity"], |
| "motion_sudden_movements": frame_data["motion_features"]["sudden_movements"], |
| } |
|
|
| |
| keypoints_data = interaction["keypoints"] |
| for prefix in ["person1_kp", "person2_kp", "relative_kp"]: |
| for i in range(17): |
| for dim in ["_x", "_y", "_conf"]: |
| row[f"{prefix}{i}{dim}"] = None |
|
|
| |
| if isinstance(keypoints_data, dict): |
| for person_prefix, kp_data in [ |
| ("person1_kp", keypoints_data.get("person1")), |
| ("person2_kp", keypoints_data.get("person2")), |
| ("relative_kp", keypoints_data.get("relative")), |
| ]: |
| if isinstance(kp_data, list): |
| for i, kp in enumerate(kp_data): |
| if i >= 17: |
| continue |
| if isinstance(kp, (list, tuple)) and len(kp) >= 3: |
| row[f"{person_prefix}{i}_x"] = float(kp[0]) |
| row[f"{person_prefix}{i}_y"] = float(kp[1]) |
| row[f"{person_prefix}{i}_conf"] = float(kp[2]) |
|
|
| return row |
|
|
