Fix plotting for msssim sampled method (#173)

* Fix plotting for msssim * Towncrier
radionets-project · Aug 16, 2024 · d40ea95 · d40ea95
1 parent 799a1aa
commit d40ea95
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/docs/changes/173.bubugfix.rst b/docs/changes/173.bubugfix.rst
@@ -0,0 +1 @@
+- fix plotting for `evaluate_msssim_sampled`
diff --git a/radionets/evaluation/train_inspection.py b/radionets/evaluation/train_inspection.py
@@ -567,7 +567,7 @@ def evaluate_ms_ssim_sampled(conf):
     name_model = Path(model_path).stem
     data_path = str(out_path) + f"/sampled_imgs_{name_model}.h5"
     loader = create_sampled_databunch(data_path, conf["batch_size"])
-    vals = []
+    vals = np.array([])
 
     # iterate trough DataLoader
     for i, (samp, std, img_true) in enumerate(tqdm(loader)):
@@ -578,10 +578,9 @@ def evaluate_ms_ssim_sampled(conf):
             win_size=7,
             size_average=False,
         )
-        vals.extend(val)
+        vals = np.append(vals, val)
 
     click.echo("\nCreating ms-ssim histogram.\n")
-    vals = torch.tensor(vals)
     histogram_ms_ssim(vals, out_path, plot_format=conf["format"])
 
     click.echo(f"\nThe mean ms-ssim value is {vals.mean()}.\n")