Scale by max value

Scale by max value to allow for processing 24- and 32-bit wavs
2022-12-16 22:12:39 -06:00 · 2022-12-16 22:12:39 -06:00 · 66dbdb16ca
parent 1c5caf77ac
commit 66dbdb16ca
1 changed files with 4 additions and 1 deletions
--- a/riffusion/audio.py
+++ b/riffusion/audio.py
@ -93,9 +93,12 @@ def image_from_spectrogram(
    """
    # Apply the power curve
    data = np.power(spectrogram, power_for_image)
+    
+    # Rescale to 0-1
+    data = data / np.max(data)

    # Rescale to 0-255
-    data = data * 255 / max_volume
+    data = data * 255

    # Invert
    data = 255 - data