fixing multifiles dataloader with resampling

Author: azrael417

makani/utils/dataloaders/data_loader_dummy.py

@@ -114,10 +114,6 @@ def __init__(self,
             torch.deg2rad(torch.tensor(self.lat_lon[1][self.crop_anchor[1] : self.crop_anchor[1] + self.crop_shape[1]])).to(torch.float32),
         )
 
-        # rescaled image shape
-        self.img_shape_resampled = (math.ceil(self.img_shape[0] / self.subsampling_factor), 
-                                    math.ceil(self.img_shape[1] / self.subsampling_factor))
-
     def _get_files_stats(self):
 
         if self.img_shape is None:
@@ -198,6 +194,9 @@ def _get_files_stats(self):
         self.img_local_offset_x = self.read_anchor[0]
         self.img_local_offset_y = self.read_anchor[1]
 
+        # resampling stuff
+        self.img_shape_resampled = (math.ceil(self.img_shape[0] / self.subsampling_factor), 
+                                    math.ceil(self.img_shape[1] / self.subsampling_factor))
         self.img_local_shape_x_resampled = self.return_shape[0]
         self.img_local_shape_y_resampled = self.return_shape[1]
         self.img_shape_x_resampled = self.img_shape_resampled[0]

makani/utils/dataloaders/data_loader_multifiles.py

@@ -153,8 +153,6 @@ def __init__(self,
             self.lat_lon_local[0][::self.subsampling_factor],
             self.lat_lon_local[1][::self.subsampling_factor],
         )
-        self.img_shape_resampled = (math.ceil(self.img_shape[0] / self.subsampling_factor), 
-                                    math.ceil(self.img_shape[1] / self.subsampling_factor))
 
         # grid types
         self.grid_converter = GridConverter(
@@ -332,6 +330,9 @@ def _get_files_stats(self, enable_logging):
         self.img_local_offset_x = self.read_anchor[0]
         self.img_local_offset_y = self.read_anchor[1]
 
+        # resampling stuff
+        self.img_shape_resampled = (math.ceil(self.img_shape[0] / self.subsampling_factor), 
+                                    math.ceil(self.img_shape[1] / self.subsampling_factor))
         self.img_local_shape_x_resampled = self.return_shape[0]
         self.img_local_shape_y_resampled = self.return_shape[1]
         self.img_shape_x_resampled = self.img_shape_resampled[0]