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

bf

Browse Source
This commit is contained in:
Till Raab 2023-10-26 10:20:19 +02:00
parent 593e8860d1
commit 6779203dc4
1 changed files with 1 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
inference.py
View File

@ -44,9 +44,7 @@ def plot_inference(img_tensor, img_name, output, target, detection_threshold):
) )
ax.set_axis_off() ax.set_axis_off()
embed() plt.savefig(Path(INFERENCE_OUTDIR)/(os.path.splitext(img_name)[0] +'_inferred.png'), dpi=IMG_DPI)
quit()
plt.savefig(Path(INFERENCE_OUTDIR)/(os.path.splitext(img_name)[0] +'_inferred.png'), IMG_DPI)
plt.close() plt.close()
# plt.show() # plt.show()