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

validation output pics now in validationaoutput/<dataset_name>

Browse Source
This commit is contained in:
Till Raab 2023-10-27 11:59:01 +02:00
parent 0befddab8d
commit a3382182e5
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] +'_predicted.png'), dpi=IMG_DPI)
plt.savefig(Path(INFERENCE_OUTDIR)/Path(DATA_DIR).name/(os.path.splitext(img_name)[0] +'_predicted.png'), dpi=IMG_DPI)
plt.close()
# plt.show()