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
