From ca352f263ab41179e95740d057e7617596d515e2 Mon Sep 17 00:00:00 2001
From: "s.islam" <s.islam@fz-juelich.de>
Date: Wed, 20 Apr 2022 12:13:18 +0200
Subject: [PATCH] Adding image grid in validation loop: To illustrate and
compare pli, cyto, and generated cyto:With Chanel(nos.) correction
---
code/data.py | 8 ++++----
code/model.py | 4 ++--
2 files changed, 6 insertions(+), 6 deletions(-)
diff --git a/code/data.py b/code/data.py
index 235cae1..618d680 100644
--- a/code/data.py
+++ b/code/data.py
@@ -122,12 +122,12 @@ class TestDataModule(pl.LightningDataModule):
# Load the pyramid/00 per file
#For JSC Training.
- pli_path = '/p/fastdata/pli/Private/oberstrass1/datasets/vervet1818/vervet1818-stained/data/aligned/pli/NTransmittance'
- cyto_path = '/p/fastdata/pli/Private/oberstrass1/datasets/vervet1818/vervet1818-stained/data/aligned/stained'
+ #pli_path = '/p/fastdata/pli/Private/oberstrass1/datasets/vervet1818/vervet1818-stained/data/aligned/pli/NTransmittance'
+ #cyto_path = '/p/fastdata/pli/Private/oberstrass1/datasets/vervet1818/vervet1818-stained/data/aligned/stained'
#For Local Machine Training.
- #pli_path = '/media/tushar/A2246889246861F1/Master Thesis MAIA/example-data/pli/NTransmittance'
- #cyto_path = '/media/tushar/A2246889246861F1/Master Thesis MAIA/example-data/stained'
+ pli_path = '/media/tushar/A2246889246861F1/Master Thesis MAIA/example-data/pli/NTransmittance'
+ cyto_path = '/media/tushar/A2246889246861F1/Master Thesis MAIA/example-data/stained'
pli_files_list = [file for file in os.listdir(pli_path) if
file.endswith(('.h5', '.hdf', '.h4', '.hdf4', '.he2', '.hdf5', '.he5'))]
diff --git a/code/model.py b/code/model.py
index 32674a3..04cd10a 100644
--- a/code/model.py
+++ b/code/model.py
@@ -53,8 +53,8 @@ class TestModule(pl.LightningModule):
loss = self.loss_f(cyto_imag_generated, batch['cyto_image'])
self.log("val_loss", loss)
if batch_idx == 0:
- grid = make_grid([batch['pli_image'][0, 0], batch['cyto_image'][0, 0], cyto_imag_generated[0, 0]])
- self.logger.experiment.add_image('Grid_images', grid, self.current_epoch, dataformats="HW")
+ grid = make_grid([batch['pli_image'][0], batch['cyto_image'][0], cyto_imag_generated[0]])
+ self.logger.experiment.add_image('Grid_images', grid, self.current_epoch, dataformats="CHW")
def configure_optimizers(self):
optimizer = torch.optim.Adam(self.parameters(), lr=self.learning_rate)
--
GitLab