From f9079bff0a6b6880a13e1662abb217e823aba1ea Mon Sep 17 00:00:00 2001
From: "Daniel.Frisinghelli" <daniel.frisinghelli@eurac.edu>
Date: Thu, 27 Aug 2020 07:45:40 +0000
Subject: [PATCH] Fixed regex: match only tif files.
---
pysegcnn/core/dataset.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/pysegcnn/core/dataset.py b/pysegcnn/core/dataset.py
index f642e05..d8dfc71 100644
--- a/pysegcnn/core/dataset.py
+++ b/pysegcnn/core/dataset.py
@@ -754,7 +754,7 @@ class StandardEoDataset(ImageDataset):
# list the spectral bands of the scene
bands = [os.path.join(dirpath, f) for f in files
- if re.search('B\\dA|B\\d{1,2}', f)]
+ if re.search('B\\dA|B\\d{1,2}(.*).tif$', f)]
# get the ground truth mask
try:
--
GitLab