Skip to content

Commit a450bf8

Browse files
committed
restructure waveplot output
1 parent 5e04bc7 commit a450bf8

1 file changed

Lines changed: 6 additions & 2 deletions

File tree

batbot/spectrogram/__init__.py

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1841,13 +1841,15 @@ def compute_wrapper(
18411841
compressed_paths = []
18421842
mask_paths = []
18431843
masked_paths = []
1844+
waveplot_compressed_paths = []
1845+
waveplot_plots = []
18441846
if not fast_mode:
18451847
datas = [
18461848
(output_paths, 'jpg', stft_db),
18471849
]
18481850
if plot_uncompressed_amplitude:
18491851
datas += [
1850-
(output_paths, 'waveplot.jpg', waveplot),
1852+
(waveplot_plots, 'waveplot.jpg', waveplot),
18511853
]
18521854
else:
18531855
datas = []
@@ -1857,7 +1859,7 @@ def compute_wrapper(
18571859
]
18581860
if 'waveplot' in segments:
18591861
datas += [
1860-
(compressed_paths, 'compressed.waveplot.jpg', segments['waveplot']),
1862+
(waveplot_compressed_paths, 'compressed.waveplot.jpg', segments['waveplot']),
18611863
]
18621864

18631865
# Create masked image
@@ -1897,6 +1899,8 @@ def compute_wrapper(
18971899
'spectrogram': {
18981900
'uncompressed.path': output_paths,
18991901
'compressed.path': compressed_paths,
1902+
'waveplot.path': waveplot_plots,
1903+
'waveplot.compressed.path': waveplot_compressed_paths,
19001904
'mask.path': mask_paths,
19011905
'masked.path': masked_paths,
19021906
},

0 commit comments

Comments
 (0)