From a450bf8f922acda17a2ef6f61abe2c2c102e7b9b Mon Sep 17 00:00:00 2001 From: Bryon Lewis Date: Mon, 16 Mar 2026 14:43:27 -0400 Subject: [PATCH 1/2] restructure waveplot output --- batbot/spectrogram/__init__.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/batbot/spectrogram/__init__.py b/batbot/spectrogram/__init__.py index 4dc583e..1545847 100644 --- a/batbot/spectrogram/__init__.py +++ b/batbot/spectrogram/__init__.py @@ -1841,13 +1841,15 @@ def compute_wrapper( compressed_paths = [] mask_paths = [] masked_paths = [] + waveplot_compressed_paths = [] + waveplot_plots = [] if not fast_mode: datas = [ (output_paths, 'jpg', stft_db), ] if plot_uncompressed_amplitude: datas += [ - (output_paths, 'waveplot.jpg', waveplot), + (waveplot_plots, 'waveplot.jpg', waveplot), ] else: datas = [] @@ -1857,7 +1859,7 @@ def compute_wrapper( ] if 'waveplot' in segments: datas += [ - (compressed_paths, 'compressed.waveplot.jpg', segments['waveplot']), + (waveplot_compressed_paths, 'compressed.waveplot.jpg', segments['waveplot']), ] # Create masked image @@ -1897,6 +1899,8 @@ def compute_wrapper( 'spectrogram': { 'uncompressed.path': output_paths, 'compressed.path': compressed_paths, + 'waveplot.path': waveplot_plots, + 'waveplot.compressed.path': waveplot_compressed_paths, 'mask.path': mask_paths, 'masked.path': masked_paths, }, From c8438bceddb553d13341b745a9e15c7db1e49e21 Mon Sep 17 00:00:00 2001 From: Bryon Lewis Date: Tue, 17 Mar 2026 08:54:18 -0400 Subject: [PATCH 2/2] prevent not including uncompressed amplitude from removing uncompressed spectrogram --- batbot/spectrogram/__init__.py | 2 -- 1 file changed, 2 deletions(-) diff --git a/batbot/spectrogram/__init__.py b/batbot/spectrogram/__init__.py index 1545847..bf51efb 100644 --- a/batbot/spectrogram/__init__.py +++ b/batbot/spectrogram/__init__.py @@ -1851,8 +1851,6 @@ def compute_wrapper( datas += [ (waveplot_plots, 'waveplot.jpg', waveplot), ] - else: - datas = [] if 'stft_db' in segments: datas += [ (compressed_paths, 'compressed.jpg', segments['stft_db']),