From e24b23957ccee5a0718f947dafda1ceaab34ab6e Mon Sep 17 00:00:00 2001
From: "Daniel.Frisinghelli" <daniel.frisinghelli@eurac.edu>
Date: Fri, 26 Feb 2021 09:18:15 +0100
Subject: [PATCH] Subdataset does not have sensor property.
---
pysegcnn/core/trainer.py | 7 ++++---
1 file changed, 4 insertions(+), 3 deletions(-)
diff --git a/pysegcnn/core/trainer.py b/pysegcnn/core/trainer.py
index 2640ca9..689f9c5 100644
--- a/pysegcnn/core/trainer.py
+++ b/pysegcnn/core/trainer.py
@@ -2593,9 +2593,10 @@ class NetworkInference(BaseConfig):
if self.plot:
# title for prediction
- title = ''.join([(v[0] + str(k)) for k, v in
- self.src_ds.sensor.band_dict().items()
- if v in self.bands])
+ title = ''.join([
+ (v[0] + str(k)) for k, v in
+ self.src_ds.dataset.sensor.band_dict().items()
+ if v in self.bands])
# plot inputs, ground truth and model predictions
fig = plot_sample(inputs.clip(0, 1),
--
GitLab