change default feature type to flair

DroneML · Jan 27, 2025 · 77835b3 · 77835b3
1 parent d5a8e04
commit 77835b3
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/segmentmytif/main.py b/src/segmentmytif/main.py
@@ -14,7 +14,7 @@
 
 
 def read_input_and_labels_and_save_predictions(input_path: Path, labels_path: Path, output_path: Path,
-                                               feature_type=FeatureType.IDENTITY, features_path: Path = None,
+                                               feature_type=FeatureType.FLAIR, features_path: Path = None,
                                                **extractor_kwargs) -> None:
     logger.info("read_input_and_labels_and_save_predictions called with the following arguments:")
     for k, v in locals().items():
@@ -120,7 +120,7 @@ def parse_args():
     parser.add_argument('-l', '--labels', type=Path, help='Path to the training labels TIFF file')
     parser.add_argument('-p', '--predictions', type=Path, help='Path to the predictions output TIFF file')
     parser.add_argument('-f', '--feature_type', type=FeatureType.from_string, choices=list(FeatureType),
-                        default=FeatureType.IDENTITY,
+                        default=FeatureType.FLAIR,
                         help='Type of feature being used. "Identity" means the raw input is directly used as features.')
 
     args = parser.parse_args()