Merge branch 'generalize_symmetry' of github.com:radionets-project/ra…

…dionets into generalize_symmetry
radionets-project · Jan 18, 2024 · 8b822cf · 8b822cf
2 parents 50aa647 + aa83722
commit 8b822cf
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 3 deletions.
diff --git a/radionets/evaluation/train_inspection.py b/radionets/evaluation/train_inspection.py
@@ -111,7 +111,7 @@ def get_prediction(conf, mode="test"):
         images["pred"] = pred
         images["indices"] = indices
 
-    if images["pred"].shape[-1] == 128:
+    if images["pred"].shape[-2] < images["pred"].shape[-1]:
         images = apply_symmetry(images)
 
     return images

diff --git a/radionets/evaluation/utils.py b/radionets/evaluation/utils.py
@@ -301,8 +301,6 @@ def get_images(test_ds, num_images, rand=False, indices=None):
 
         img_test = test_ds[indices][0]
         img_true = test_ds[indices][1]
-        img_test = img_test[:, :, :65, :]
-        img_true = img_true[:, :, :65, :]
         return img_test, img_true, indices
     else:
         mean = test_ds[indices][0]