From 47d202ba5d8b9a57d39594602e189caddd93ffb6 Mon Sep 17 00:00:00 2001
From: Nassim Bouteldja <nbouteldja@ukaachen.de>
Date: Tue, 5 Jul 2022 00:02:39 +0200
Subject: [PATCH] Update postprocessing.py
---
postprocessing.py | 8 ++++----
1 file changed, 4 insertions(+), 4 deletions(-)
diff --git a/postprocessing.py b/postprocessing.py
index 40933ce..970f1cd 100644
--- a/postprocessing.py
+++ b/postprocessing.py
@@ -7,7 +7,7 @@ import math
from scipy.ndimage.measurements import label
from scipy.ndimage.morphology import binary_dilation, binary_fill_holes
-from utils import getChannelSmootingConvLayer
+# from utils import getChannelSmootingConvLayer
structure = np.zeros((3, 3), dtype=np.int)
@@ -85,9 +85,9 @@ def postprocessPredictionAndGT(prediction, GT, device, predictionsmoothing, hole
2.network output prediction (w/o postprocessing)
"""
################# PREDICTION SMOOTHING ################
- if predictionsmoothing:
- smoothingKernel = getChannelSmootingConvLayer(8).to(device)
- prediction = smoothingKernel(prediction)
+ #if predictionsmoothing:
+ # smoothingKernel = getChannelSmootingConvLayer(8).to(device)
+ # prediction = smoothingKernel(prediction)
# labelMap contains following labels: 0/1/2/3/4/5/6/7 => Background/tubuli/glom_full/glom_tuft/veins/artery_full/artery_lumen/border
labelMap = torch.argmax(prediction, dim=1).squeeze(0).to("cpu").numpy() # Label 0/1/2/3/4/5/6/7: Background/tubuli/glom_full/glom_tuft/veins/artery_full/artery_lumen/border
--
GitLab