From a3382182e58d393e305627230b5bef7cc48fb332 Mon Sep 17 00:00:00 2001
From: Till Raab <till.raab@uni-tuebingen.de>
Date: Fri, 27 Oct 2023 11:59:01 +0200
Subject: [PATCH] validation output pics now in
 validationaoutput/<dataset_name>

---
 train.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/train.py b/train.py
index 491149a..4a99a3e 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] +'_predicted.png'), dpi=IMG_DPI)
+    plt.savefig(Path(INFERENCE_OUTDIR)/Path(DATA_DIR).name/(os.path.splitext(img_name)[0] +'_predicted.png'), dpi=IMG_DPI)
     plt.close()
     # plt.show()