changes for multichannel, small adjustments
This commit is contained in:
20
plot.py
20
plot.py
@@ -14,22 +14,26 @@ parser = argparse.ArgumentParser()
|
||||
parser.add_argument('--infile', type = str, required = True, help = 'file name of the .h5 file for names')
|
||||
parser.add_argument('--data', type = str, required = True, help = "write only the part of 1 file name before the _(nr-s).npy; files for creating the plots")
|
||||
parser.add_argument('--images', type = str, default = "mse_mssim", help = "naming the output images, default = mse_mssim")
|
||||
parser.add_argument('-q','--squared', action = argparse.BooleanOptionalAction, help = "squares the intensity values, otherwise normal values")
|
||||
parser.add_argument('-sp', '--sigma_plots', action = argparse.BooleanOptionalAction, help = 'shows the plots, where all intensity values of a sigma value are on one plot')
|
||||
parser.add_argument('-q','--squared', action = argparse.BooleanOptionalAction, help = "squares the intensity values, otherwise normal values")
|
||||
|
||||
args = parser.parse_args()
|
||||
|
||||
group_to_datasets = {}
|
||||
|
||||
with h5py.File(args.infile, 'r') as hdf5_file:
|
||||
for group_name in hdf5_file: # loop üle grupi nimede
|
||||
group = hdf5_file[group_name] #salvestab grupi nime
|
||||
for group_name in hdf5_file: # salvestab grupi nime
|
||||
group = hdf5_file[group_name]
|
||||
if isinstance(group, h5py.Group):
|
||||
datasets = [] #kui grupi nimi on h5py.Group nimi siis
|
||||
for ds_name in group: #vaatab üle kõik datasetid grupi sees
|
||||
if isinstance(group[ds_name], h5py.Dataset): # kui vastab ds nimele
|
||||
datasets.append(ds_name) # appenditakse
|
||||
group_to_datasets[group_name] = datasets # iga grupile apenditakse tema oma ds
|
||||
for subgroup_name in group: # salvestab subgrupi nime
|
||||
subgroup = group[subgroup_name]
|
||||
if isinstance(subgroup, h5py.Group):
|
||||
datasets = [] #kui grupi nimi on h5py.Group nimi siis
|
||||
for ds_name in subgroup:#vaatab üle kõik datasetid grupi sees
|
||||
if isinstance(subgroup[ds_name], h5py.Dataset):# kui vastab ds nimele
|
||||
datasets.append(ds_name)# appenditakse
|
||||
group_to_datasets[subgroup_name] = datasets # iga grupile apenditakse tema oma ds
|
||||
# added lines for recognising groups and subgroups
|
||||
|
||||
group_names = list(group_to_datasets.keys()) # has all the groups which are different inensitires and
|
||||
# all the datasets names are there with sigma and minimum but i only use sigma so the rest are not used at all
|
||||
|
||||
Reference in New Issue
Block a user