structure

data/generate_dataset.py

@@ -10,7 +10,6 @@ from pathlib import Path
 from tqdm.auto import tqdm
 
 import itertools
-
 import sys
 import os
 
@@ -63,7 +62,7 @@ def main(folder):
         s_trans = transformed(log_s.unsqueeze(0))
 
         fig_title = (f'{Path(folder).name}__{t0:.0f}s-{t1:.0f}s__{f0:.0f}-{f1:.0f}Hz').replace(' ', '0')
-        fig = plt.figure(figsize=(10, 7), num=fig_title)
+        fig = plt.figure(figsize=(7, 7), num=fig_title)
         gs = gridspec.GridSpec(1, 2, width_ratios=(8, 1), wspace=0)# , bottom=0, left=0, right=1, top=1
         gs2 = gridspec.GridSpec(1, 1, bottom=0, left=0, right=1, top=1)#
         ax = fig.add_subplot(gs2[0, 0])
@@ -74,7 +73,7 @@ def main(folder):
         # fig.colorbar(im, cax=cax)
         ax.axis(False)
 
-        plt.savefig(fig_title + '.png', dpi=300)
+        plt.savefig(fig_title + '.png', dpi=256)
         plt.close()
     # # ax.imshow(spec[f0:f1, t0:t1], cmap='gray')