Update biomassters.py

VMarsocci · Sep 22, 2024 · 7461e2d · 7461e2d
1 parent 049964c
commit 7461e2d
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/datasets/biomassters.py b/datasets/biomassters.py
@@ -10,10 +10,10 @@
 
 def read_imgs(multi_temporal, temp , fname, data_dir, img_size):
     imgs_s1, imgs_s2, mask = [], [], []
-    if multi_temporal:
-        month_list = list(range(12))
+    if multi_temporal==1:
+        month_list = [temp]        
     else:
-        month_list = [temp]
+        month_list = list(range(12))
 
     for month in month_list: