From 6779203dc48a8b5956ebb13b2ea6911e6fba38e1 Mon Sep 17 00:00:00 2001
From: Till Raab <till.raab@uni-tuebingen.de>
Date: Thu, 26 Oct 2023 10:20:19 +0200
Subject: [PATCH] bf

---
 inference.py | 4 +---
 1 file changed, 1 insertion(+), 3 deletions(-)

diff --git a/inference.py b/inference.py
index 945f65e..f9b90a1 100644
--- a/inference.py
+++ b/inference.py
@@ -44,9 +44,7 @@ def plot_inference(img_tensor, img_name, output, target, detection_threshold):
         )
 
     ax.set_axis_off()
-    embed()
-    quit()
-    plt.savefig(Path(INFERENCE_OUTDIR)/(os.path.splitext(img_name)[0] +'_inferred.png'), IMG_DPI)
+    plt.savefig(Path(INFERENCE_OUTDIR)/(os.path.splitext(img_name)[0] +'_inferred.png'), dpi=IMG_DPI)
     plt.close()
     # plt.show()