@@ -18,7 +18,7 @@ def repro_equidistant_centers(
1818):
1919 default_cell_type = "ihc"
2020 default_component_list = [1 ]
21- default_halo_size = [256 , 256 , 50 ]
21+ default_halo_size = [256 , 256 , 128 ]
2222 default_n_blocks = 6
2323
2424 with open (ddict , 'r' ) as myfile :
@@ -30,6 +30,14 @@ def repro_equidistant_centers(
3030 if os .path .isfile (output_path ) and not force_overwrite :
3131 print (f"Skipping { output_path } . File already exists." )
3232
33+ def update_dic (dic , keyword , default ):
34+ if keyword in dic :
35+ value = dic [keyword ]
36+ else :
37+ value = default
38+ dic [keyword ] = value
39+ return value
40+
3341 for dic in param_dicts :
3442 cochlea = dic ["cochlea" ]
3543 seg_channel = dic ["segmentation_channel" ]
@@ -42,17 +50,16 @@ def repro_equidistant_centers(
4250 with fs .open (tsv_path , 'r' ) as f :
4351 table = pd .read_csv (f , sep = "\t " )
4452
45- cell_type = dic [ "type" ] if "type" in dic else default_cell_type
46- component_list = dic [ "component_list" ] if "component_list" in dic else default_component_list
47- halo_size = dic [ "halo_size" ] if "halo_size" in dic else default_halo_size
48- n_blocks = dic [ "n_blocks" ] if "n_blocks" in dic else default_n_blocks
53+ cell_type = update_dic ( dic , "type" , default_cell_type )
54+ component_list = update_dic ( dic , "component_list" , default_component_list )
55+ _ = update_dic ( dic , "halo_size" , default_halo_size )
56+ n_blocks = update_dic ( dic , "n_blocks" , default_n_blocks )
4957
5058 centers = equidistant_centers (table , component_label = component_list , cell_type = cell_type , n_blocks = n_blocks )
5159 centers = [[int (c ) for c in center ] for center in centers ]
52- ddict = dic .copy ()
53- ddict ["crop_centers" ] = centers
54- ddict ["halo_size" ] = halo_size
55- out_dict .append (ddict )
60+
61+ dic ["crop_centers" ] = centers
62+ out_dict .append (dic )
5663
5764 with open (output_path , "w" ) as f :
5865 json .dump (out_dict , f , indent = '\t ' , separators = (',' , ': ' ))
0 commit comments