From 5ad32ccccfb282591dfd12f6f0365cdc91ad6545 Mon Sep 17 00:00:00 2001
From: "Daniel.Frisinghelli" <daniel.frisinghelli@eurac.edu>
Date: Thu, 4 Feb 2021 17:52:39 +0100
Subject: [PATCH] Fixed a bug: Read dataset not Subset
---
pysegcnn/core/trainer.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/pysegcnn/core/trainer.py b/pysegcnn/core/trainer.py
index d09a6c3..9105af4 100644
--- a/pysegcnn/core/trainer.py
+++ b/pysegcnn/core/trainer.py
@@ -2546,7 +2546,7 @@ class NetworkInference(BaseConfig):
self.bands,
self.source_labels,
y=labels,
- y_pred={self.model.__class__.__name__: prdctn},
+ y_pred={model.__class__.__name__: prdctn},
accuracy=True,
state=batch_name,
plot_path=self.scenes_path,
--
GitLab