dataset_utils.py

from gen_utils import read_json

def load_motionfix(path_to_data):
    dataset = read_json(path_to_data + '/amt_motionfix_latest.json')
    splits = read_json(path_to_data + '/splits.json')

    train_data = {}
    val_data = {}
    test_data = {}

    # Fill each dictionary with the corresponding data
    for key in splits['train']:
        train_data[key] = dataset[key]

    for key in splits['val']:
        val_data[key] = dataset[key]

    for key in splits['test']:
        test_data[key] = dataset[key]
    validation_test_data = {**val_data, **test_data}

    return train_data, validation_test_data