utils.py 文件源码

def create_ds_from_splits(mat_file_prefix, db, splits):
    """Create datasets from splits of data.

    This function is created because of splits from larger datasets
    """
    for batch_idx in xrange(1, splits+1):
        # load data
        temp_data = load_mat_data(mat_file_prefix+str(batch_idx)+".mat")
        print ("[MESSAGE] Loaded %d-th split" % batch_idx)

        group_name = "grid_data_split_"+str(batch_idx)

        # save data within splits
        for key in data_dict:
            add_h5_group(group_name, db)
            add_h5_ds(temp_data[key], key, db, group_name)
            print ("[MESSAGE] %s: Saved %s" % (group_name, key))