lc 9 should be 7 channel

nasa-nccs-hpda · Jul 12, 2024 · 9066446 · 9066446
1 parent 182f7a7
commit 9066446
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/pytorch_caney/data/datasets/modis_lc_nine_dataset.py b/pytorch_caney/data/datasets/modis_lc_nine_dataset.py
@@ -53,8 +53,7 @@ def __len__(self):
     def __getitem__(self, idx, transpose=True):
 
         # load image
-        img = np.random.rand(224, 224, 14)
-        # img = np.load(self.img_list[idx])
+        img = np.load(self.img_list[idx])
 
         # load mask
         mask = np.load(self.mask_list[idx])