From a6641accaf3cf285dbfc16ac00afde83df3fd72f Mon Sep 17 00:00:00 2001
From: Till Raab <till.raab@uni-tuebingen.de>
Date: Fri, 27 Oct 2023 09:35:29 +0200
Subject: [PATCH] infered img not with suffix: _predicted.png
---
train.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/train.py b/train.py
index db2c584..491149a 100644
--- a/train.py
+++ b/train.py
@@ -111,7 +111,7 @@ def plot_validation(img_tensor, img_name, output, target, detection_threshold):
)
ax.set_axis_off()
- plt.savefig(Path(INFERENCE_OUTDIR)/(os.path.splitext(img_name)[0] +'_inferred.png'), dpi=IMG_DPI)
+ plt.savefig(Path(INFERENCE_OUTDIR)/(os.path.splitext(img_name)[0] +'_predicted.png'), dpi=IMG_DPI)
plt.close()
# plt.show()