Function eval_Div_given_paths

evaluation/patch_utils.py:169–181

(data_paths, device="cpu")

def eval_Div_given_paths(data_paths, device="cpu"):
    random.seed(1234)

    gen_data_list = []
    for path in data_paths:
        gen_data = load_voxgrid(path, resolution=128, device=device)
        gen_data_list.append(gen_data)

    gen_data_list = torch.stack(gen_data_list, dim=0)
    div = pairwise_IoU_dist(gen_data_list).round(6)
    
    eval_results = {'Div': div}
    return eval_results