From 1beaf69be374eedb5aa9006ce3ee099f93915ab4 Mon Sep 17 00:00:00 2001
From: "Daniel.Frisinghelli" <daniel.frisinghelli@eurac.edu>
Date: Thu, 30 Jul 2020 16:45:10 +0200
Subject: [PATCH] Moved accuracy function to utils.py
---
pysegcnn/core/utils.py | 9 +++++++++
1 file changed, 9 insertions(+)
diff --git a/pysegcnn/core/utils.py b/pysegcnn/core/utils.py
index 61b69e8..8be2d19 100644
--- a/pysegcnn/core/utils.py
+++ b/pysegcnn/core/utils.py
@@ -11,6 +11,7 @@ import datetime
# externals
import gdal
+import torch
import numpy as np
# the following functions are utility functions for common image
@@ -296,6 +297,14 @@ def reconstruct_scene(tiles, img_size, tile_size=None, nbands=1):
return scene.squeeze()
+# function calculating prediction accuracy
+def accuracy_function(outputs, labels):
+ if isinstance(outputs, torch.Tensor):
+ return (outputs == labels).float().mean().item()
+ else:
+ return (np.asarray(outputs) == np.asarray(labels)).mean().item()
+
+
def parse_landsat_scene(scene_id):
# Landsat Collection 1 naming convention in regular expression
--
GitLab