changes for multichannel, small adjustments

This commit is contained in:
2025-10-20 16:17:28 +03:00
parent 5c1077f3d7
commit 112ae6d5b0
2 changed files with 77 additions and 53 deletions

20
plot.py
View File

@@ -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