raab/efishSignalDetector
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

embed for figsave bf

Browse Source
This commit is contained in:
Till Raab 2023-10-26 10:14:44 +02:00
parent 76cee2c050
commit 593e8860d1
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
inference.py
View File

@ -44,6 +44,8 @@ def plot_inference(img_tensor, img_name, output, target, detection_threshold):
) )
ax.set_axis_off() 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'), IMG_DPI)
plt.close() plt.close()
# plt.show() # plt.show()