def web_length_metric(gold_swc, test_swc, method, rad_threshold, len_threshold): gold_tree = SwcTree() test_tree = SwcTree() gold_tree.load_list(adjust_swcfile(gold_swc)) test_tree.load_list(adjust_swcfile(test_swc)) config = { 'method': method, 'len_threshold': len_threshold, 'rad_threshold': rad_threshold } recall, precision, vertical_tree = length_metric(gold_swc_tree=gold_tree, test_swc_tree=test_tree, abs_dir="", config=config) # gold_tree.radius_limit(10) # test_tree.radius_limit(10) result = { 'recall': recall, 'precision': precision, 'gold_swc': swc_to_list(gold_tree), 'test_swc': swc_to_list(test_tree), 'vertical_swc': vertical_tree } return result
def web_diadem_metric(gold_swc, test_swc, config): gold_tree = SwcTree() test_tree = SwcTree() gold_tree.load_list(adjust_swcfile(gold_swc)) test_tree.load_list(adjust_swcfile(test_swc)) diadem_metric(swc_gold_tree=gold_tree, swc_test_tree=test_tree, config=config) # gold_tree.radius_limit(10) # test_tree.radius_limit(10) print(swc_to_list(test_tree)) result = { 'gold_swc': swc_to_list(gold_tree), 'test_swc': swc_to_list(test_tree), 'weight_sum': g_weight_sum, 'score_sum': g_score_sum, 'final_score': g_final_score } return result
def read_from_str(swc_str): swc_tree = SwcTree() swc_list = adjust_swcfile(swc_str) swc_tree.load_list(swc_list) return swc_tree