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

infered img not with suffix: _predicted.png

Browse Source
This commit is contained in:
Till Raab 2023-10-27 09:35:29 +02:00
parent 1042e44f37
commit a6641accaf
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
train.py
View File

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