manage transforms

mpelchat04 · Oct 7, 2024 · e8c9e0c · e8c9e0c
1 parent 23a8589
commit e8c9e0c
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/inference_segmentation.py b/inference_segmentation.py
@@ -67,6 +67,7 @@ def main(params:Union[DictConfig, Dict]):
     vectorize = get_key_def('ras2vec', params['inference'], expected_type=bool, default=False)
     transform_flip = get_key_def('flip', params['inference'], expected_type=bool, default=False)
     transform_rotate = get_key_def('rotate', params['inference'], expected_type=bool, default=False)
+    transforms = True if transform_flip or transform_rotate else False
 
     if raw_data_csv and input_stac_item:
         raise ValueError(f"Input imagery should be either a csv of stac item. Got inputs from both \"raw_data_csv\" "
@@ -108,7 +109,7 @@ def main(params:Union[DictConfig, Dict]):
                                  device=device_str,
                                  gpu_id=gpu_index,
                                  prediction_threshold=prediction_threshold,
-                                 transformers=True,
+                                 transformers=transforms,
                                  transformer_flip=transform_flip,
                                  transformer_rotate=transform_rotate,
                                  )