Function convert_tree_json

code_generation_benchmark/utils/utils_eval.py:233–257

(logger, tree_file, file_type)

        gt_match_sims[i] = sims_record[i][j]

def convert_tree_json(logger, tree_file, file_type):
    assert file_type in ['gt', 'pr'], 'file_type should be gt or pr'
    logger.info(f'converting {tree_file} to leafs')
    tree_json = load_json(tree_file)
    ref_bbox = tree_json['bounding_box']
    normalize_tree_leafs(tree_json, ref_bbox)
    leafs = parse_all_leafs(tree_json, [])
    logger.info(f'*** {file_type}    all: {len(leafs)}')

    for leaf in leafs:
        leaf['valid'] = judge_invalid_elements(leaf)
        if file_type == 'pr' and leaf['node']['tag'] == 'image':
            leaf['valid'] = False
    safe_save_json(leafs, tree_file.replace('.json', '_ori_leafs.json'))

    leafs = [leaf for leaf in leafs if leaf['valid']]
    logger.info(f'*** {file_type} filter: {len(leafs)}')
    # reorder leafs by x and y and update ids
    leafs = sorted(leafs, key=lambda x: x['node']['norm_bbox'][0])
    leafs = sorted(leafs, key=lambda x: x['node']['norm_bbox'][1])
    for i, leaf in enumerate(leafs):
        leaf['id'] = i
    safe_save_json(leafs, tree_file.replace('.json', '_leafs.json'))
    leaf_bboxes = [leaf['node']['norm_bbox'] for leaf in leafs]
    vis_bboxes_with_indexes(leaf_bboxes, tree_file.replace('.json', '.png'), tree_file.replace('.json', '_leafs.png'))

def compute_element_pairs(logger, gt_file, pr_file):
    gt_leafs_file = gt_file.replace('.json', '_leafs.json')