Function compute_jitter_degree

mbench/motion_quality.py:127–170

Compute the jitter degree of the motion based on the acceleration of the joints.

(full_info_path: str, device: str, **kwargs)

    }

def compute_jitter_degree(full_info_path: str, device: str, **kwargs):
    """
    Compute the jitter degree of the motion based on the acceleration of the joints.
    """
    prompt_dict_ls = load_dimension_info(full_info_path, dimension='Jitter_Degree')
    
    jitter_degree_list = []
    per_motion_metrics = []
    
    for prompt_dict in tqdm(prompt_dict_ls):
        evaluation_file = prompt_dict["evaluation_file"]
        pred_joints = load_joints(evaluation_file, device)

        # Global jitter degree
        velocity = pred_joints[1:] - pred_joints[:-1]  # Shape: (T-1, 24, 3)
        acceleration = velocity[1:] - velocity[:-1]  # Shape: (T-2, 24, 3)
        acceleration_magnitude = torch.norm(acceleration, dim=2)  # Shape: (T-2, 24)
        global_jitter = acceleration_magnitude.mean()
        
        # Local jitter degree (remove global translation)
        local_joints = remove_global_translation(pred_joints)
        local_velocity = local_joints[1:] - local_joints[:-1]  # Shape: (T-1, 24, 3)
        local_acceleration = local_velocity[1:] - local_velocity[:-1]  # Shape: (T-2, 24, 3)
        local_acceleration_magnitude = torch.norm(local_acceleration, dim=2)  # Shape: (T-2, 24)
        local_jitter = local_acceleration_magnitude.mean()
        
        # Combined jitter degree
        combined_jitter = global_jitter + local_jitter
        combined_value = combined_jitter.item()
        jitter_degree_list.append(combined_value)
        per_motion_metrics.append(
            {
                "id": prompt_dict.get("id"),
                "prompt": prompt_dict.get("prompt"),
                "value": combined_value,
                "evaluation_file": evaluation_file,
                "motion_duration": prompt_dict.get("motion_duration"),
            }
        )
    
    return {
        "aggregate": summarize_scores(jitter_degree_list),
        "per_motion": per_motion_metrics,
    }

def compute_ground_penetration(full_info_path: str, device: str, **kwargs):
    """