Function _load_test_indices

deeplabcut/benchmark/metrics.py:256–260

Returns the indices of test images in the training dataset dataframe.

(shuffle_metadata_path: str)

def _load_test_indices(shuffle_metadata_path: str) -> list[int]:
    """Returns the indices of test images in the training dataset dataframe."""
    with open(shuffle_metadata_path, "rb") as f:
        test_indices = set([int(i) for i in pickle.load(f)[2]])
    return list(sorted(test_indices))