create dateset update

2023-10-20 09:00:28 +02:00 · 2023-10-20 09:00:28 +02:00 · 7a8790d9af
commit 7a8790d9af
parent cd3a066427
1 changed files with 1 additions and 1 deletions
--- 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')