def output_transform_va(process_output):
    """
    Output transform for vehicle affordances metrics.
    """
    y_pred = process_output[0]['vehicle_affordances'].argmax(dim=1)
    y = process_output[1]['vehicle_affordances'].argmax(dim=1)
    return dict(y_pred=y_pred, y=y)