experanto.utils.add_behavior_as_channels

add_behavior_as_channels(data)[source]

Add behavioral data as additional channels to screen data.

Parameters:

data (dict) – Dictionary with keys ‘screen’, ‘eye_tracker’, ‘treadmill’. Screen shape: (C, T, H, W). Behavior shapes: (T, C_behavior) or (T, H, W).

Returns:

Modified dictionary with behavior concatenated to screen channels. Screen shape becomes (C + behavior_channels, T, H, W).

Return type:

dict