From 593e8860d175587de8a84a7f1c176778cb1f6949 Mon Sep 17 00:00:00 2001 From: Till Raab Date: Thu, 26 Oct 2023 10:14:44 +0200 Subject: [PATCH] embed for figsave bf --- inference.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/inference.py b/inference.py index 619b9b3..945f65e 100644 --- a/inference.py +++ b/inference.py @@ -44,6 +44,8 @@ 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.close() # plt.show()