deepof.utils.iterative_imputation

deepof.utils.iterative_imputation(project: deepof_project, tab_dict: dict, lik_dict: dict, full_imputation: bool = False)

Perform iterative imputation on occluded body parts. Run per animal and experiment.

Parameters:

• project (project) – Project object.

• tab_dict (dict) – Dictionary with the coordinates of the body parts.

• lik_dict (dict) – Dictionary with the likelihood of the tracking for each body part and animal.

• full_imputation (bool) – Determines if only small gaps get linearily imputed (False) or additionally IterativeImputer and a few other steps are executed to close all gaps (True)

Returns:

Dictionary with the coordinates of the body parts after imputation.

Return type:

tab_dict (dict)