Adding image grid in validation loop: To illustrate and compare pli, cyto, and...

Adding image grid in validation loop: To illustrate and compare pli, cyto, and generated cyto:With Chanel(nos.) correction

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'))]

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)