From 7a8790d9af7f1d2112af6af05b8e2569b2d8034d Mon Sep 17 00:00:00 2001 From: Till Raab Date: Fri, 20 Oct 2023 09:00:28 +0200 Subject: [PATCH] create dateset update --- data/generate_dataset.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/data/generate_dataset.py b/data/generate_dataset.py index 7631203..e3e7aab 100644 --- a/data/generate_dataset.py +++ b/data/generate_dataset.py @@ -73,7 +73,7 @@ def main(folder): # fig.colorbar(im, cax=cax) ax.axis(False) - plt.savefig(fig_title + '.png', dpi=256) + plt.savefig(os.path.join('train', fig_title + '.png'), dpi=256) plt.close() # # ax.imshow(spec[f0:f1, t0:t1], cmap='gray')