Add files via upload

ytao-wang · Sep 19, 2024 · 45347c5 · 45347c5
1 parent 0bb40fd
commit 45347c5
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/train_cvact.py b/train_cvact.py
@@ -36,6 +36,7 @@ class Configuration:
 
     # is use polar
     is_polar: bool = False
+    psm: bool = True
     image_size_sat = (256, 256)
     img_size_ground = (128, 512)
 
@@ -147,7 +148,7 @@ class Configuration:
     model = TimmModel(config.model,
                       config.image_size_sat,
                       config.img_size_ground,
-                      psm=True,
+                      psm=config.psm,
                       is_polar=config.is_polar)
 
     mean=[0.485, 0.456, 0.406]

diff --git a/train_cvusa.py b/train_cvusa.py
@@ -34,6 +34,7 @@ class Configuration:
 
     # is use polar
     is_polar: bool = False
+    psm: bool = True
     image_size_sat = (256, 256)
     img_size_ground = (128, 512)
 
@@ -143,7 +144,7 @@ class Configuration:
     model = TimmModel(config.model,
                       config.image_size_sat,
                       config.img_size_ground,
-                      psm=True,
+                      psm=config.psm,
                       is_polar=config.is_polar)
 
     mean=[0.485, 0.456, 0.406]