From 9a7264c8c690cad2045060bd8a4564e9d07f1a46 Mon Sep 17 00:00:00 2001
From: "Daniel.Frisinghelli" <daniel.frisinghelli@eurac.edu>
Date: Wed, 3 Feb 2021 14:19:59 +0100
Subject: [PATCH] Preparing test model runs.
---
pysegcnn/main/train_config.py | 15 ++++++++-------
1 file changed, 8 insertions(+), 7 deletions(-)
diff --git a/pysegcnn/main/train_config.py b/pysegcnn/main/train_config.py
index 8050631..d83a6f2 100644
--- a/pysegcnn/main/train_config.py
+++ b/pysegcnn/main/train_config.py
@@ -29,25 +29,26 @@ import pathlib
HERE = pathlib.Path(__file__).resolve().parent
# 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'),
- 'Alcd': DRIVE_PATH.joinpath('Alcd/60m')
+ 'Alcd': DRIVE_PATH.joinpath('Alcd')
}
# name of the dataset
DS_NAME = 'Sparcs'
# spectral bands to use for training
-BANDS = ['red', 'green', 'blue', 'nir', 'swir1', 'swir2']
+BANDS = ['red', 'green', 'blue']
# tile size of a single sample
-TILE_SIZE = None
+TILE_SIZE = 128
# number of folds for cross validation
-K_FOLDS = 1
+K_FOLDS = 10
# the source dataset configuration dictionary
ds_config = {
@@ -249,10 +250,10 @@ model_config = {
'early_stop': True,
'mode': 'max',
'delta': 0,
- 'patience': 10,
+ 'patience': 20,
# define the number of epochs: the number of maximum iterations over
# the whole training dataset
- 'epochs': 100,
+ 'epochs': 5000,
}
--
GitLab