From 46205d1e3b6c3a2e3a9c2d2aae0bfbc381884282 Mon Sep 17 00:00:00 2001
From: "Daniel.Frisinghelli" <daniel.frisinghelli@eurac.edu>
Date: Wed, 3 Mar 2021 14:26:05 +0100
Subject: [PATCH] Added settings for Sparcs target dataset.
---
pysegcnn/main/eval_config.py | 24 ++++++++++++++----------
1 file changed, 14 insertions(+), 10 deletions(-)
diff --git a/pysegcnn/main/eval_config.py b/pysegcnn/main/eval_config.py
index da215db..f7907fd 100644
--- a/pysegcnn/main/eval_config.py
+++ b/pysegcnn/main/eval_config.py
@@ -20,8 +20,9 @@ License
import pathlib
# path to the datasets on the current machine
-DRIVE_PATH = pathlib.Path('C:/Eurac/Projects/CCISNOW/Datasets/')
+# DRIVE_PATH = pathlib.Path('C:/Eurac/Projects/CCISNOW/Datasets/')
# DRIVE_PATH = pathlib.Path('/mnt/CEPH_PROJECTS/cci_snow/dfrisinghelli/Datasets/') # nopep8
+DRIVE_PATH = pathlib.Path('/home/dfrisinghelli/Datasets')
# name and paths to the datasets
DATASETS = {'Sparcs': DRIVE_PATH.joinpath('Sparcs'),
@@ -29,26 +30,29 @@ DATASETS = {'Sparcs': DRIVE_PATH.joinpath('Sparcs'),
}
# name of the target dataset
-TRG_DS = 'Alcd'
+TRG_DS = 'Sparcs'
# spectral bands to use for evaluation
-BANDS = ['red', 'green', 'blue', 'nir', 'swir1', 'swir2']
+BANDS = ['red', 'green', 'blue']
# tile size of a single sample
-TILE_SIZE = 128
+TILE_SIZE = 64
# the target dataset configuration dictionary
trg_ds = {
- 'dataset_name': 'Sparcs',
+ 'dataset_name': TRG_DS,
'root_dir': DATASETS[TRG_DS],
- 'gt_pattern': '(.*)Labels\\.tif',
+ #'gt_pattern': '(.*)Labels\\.tif',
+ 'gt_pattern': '(.*)mask\\.png',
'bands': BANDS,
'tile_size': TILE_SIZE,
'pad': True,
- 'sort': True,
+ 'sort': False,
'transforms': [],
- 'merge_labels': {'Cirrus': 'Cloud',
- 'Not_used': 'No_data'}
+ # 'merge_labels': {'Cirrus': 'Cloud',
+ # 'Not_used': 'No_data'}
+ 'merge_labels': {'Shadow_over_water': 'Shadow',
+ 'Flooded': 'Land'}
}
@@ -61,6 +65,6 @@ trg_ds_split = {
'seed': 0,
'shuffle': True,
'ttratio': 1,
- 'tvratio': 0.8,
+ 'tvratio': 0.95,
}
--
GitLab