embed for figsave bf

2023-10-26 10:14:44 +02:00 · 2023-10-26 10:14:44 +02:00 · 593e8860d1
commit 593e8860d1
parent 76cee2c050
1 changed files with 2 additions and 0 deletions
--- 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()