diff --git a/pysegcnn/core/trainer.py b/pysegcnn/core/trainer.py index 5a3bf3a90d1e07ec859d2a63d2b1bc74deaa201f..8b461945466b7bacd300618cb2ddc420eccdc367 100644 --- a/pysegcnn/core/trainer.py +++ b/pysegcnn/core/trainer.py @@ -2074,7 +2074,7 @@ class NetworkInference(BaseConfig): predict_scene: bool = True plot_scenes: bool = False plot_bands: list = dataclasses.field( - default_factory=lambda: ['nir', 'red', 'green']) + default_factory=lambda: ['swir2', 'nir', 'green']) cm: bool = True figsize: tuple = (10, 10) alpha: int = 5