cell_segmentation/eval_loop.py at main · MaxLevianSterling/cell_segmentation · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
from eval import binary2json
from eval import evaluate


network_names = ['D','E','F','G','H','I','J','K','L','M','N','O','P','Q','R','S','T','U','V','W']

for duplicate in range(3):
    for iN, network_name in enumerate(network_names):
        binary2json(
            path = '/mnt/sdg/maxs',
            data_set = 'LIVECell',
            data_type = 'part_set',
            data_subset = '5',
            subset_type = 'test',
            annot_type = 'soma',
            mode = 'eval',
            model=f'{network_names[iN]}{duplicate}',
            checkpoint = 1000,
            print_separator = '$',
        )
        #try:
        evaluate(
            path = '/mnt/sdg/maxs',
            data_set = 'LIVECell',
            data_type = 'part_set',
            data_subset = '5',
            subset_type = 'test',
            annot_type = 'soma',
            mode = 'original',
            model=f'{network_names[iN]}{duplicate}',
            checkpoint = 1000,
            print_separator = '$',
        )
        #except:
        #    pass