From 70742a1d8b9f12c4faf25e4ddb60d0ab7921df77 Mon Sep 17 00:00:00 2001 From: Siddharth Kekre Date: Tue, 2 Jan 2024 00:10:38 +0000 Subject: [PATCH] Changes AutoEncoder Hidden Layers --- Code/autoencoder_model.py | 4 ++-- Code/main.py | 4 ++-- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/Code/autoencoder_model.py b/Code/autoencoder_model.py index 975d07e..c4e34d2 100644 --- a/Code/autoencoder_model.py +++ b/Code/autoencoder_model.py @@ -16,9 +16,9 @@ class Grey2RGBAutoEncoder(nn.Module): def __init__(self): super(Grey2RGBAutoEncoder, self).__init__() # Define the Encoder - self.encoder = self._make_layers([1, 64, 128, 256]) + self.encoder = self._make_layers([1, 8, 16, 32]) # Define the Decoder - self.decoder = self._make_layers([256, 128, 64, 3], decoder=True) + self.decoder = self._make_layers([32, 16, 8, 3], decoder=True) # Helper function to create the encoder or decoder layers. def _make_layers(self, channels, decoder=False): diff --git a/Code/main.py b/Code/main.py index 5dc44bd..e09b6aa 100644 --- a/Code/main.py +++ b/Code/main.py @@ -22,8 +22,8 @@ rgb_dir = '../Dataset/RGB' # Define Universal Parameters -image_height = 400 -image_width = 600 +image_height = 3000 +image_width = 4500 batch_size = 2 def main_worker(rank, world_size):