Mirrors
/
riffusion-inference
mirror of https://github.com/riffusion/riffusion.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Convert to three-channel if needed in spectrogram_from_image 

Closes: #51

Topic: convert_three_channel
This commit is contained in:
Hayk Martiros 2022-12-29 11:27:42 -08:00
parent 89f3946155
commit ce8744897b
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
riffusion/util/image_util.py
View File

@ -77,6 +77,10 @@ def spectrogram_from_image(
Returns: Returns:
spectrogram: (channels, frequency, time) spectrogram: (channels, frequency, time)
""" """
# Convert to RGB if single channel
if image.mode in ("P", "L"):
image = image.convert("RGB")
# Flip Y # Flip Y
image = image.transpose(Image.Transpose.FLIP_TOP_BOTTOM) image = image.transpose(Image.Transpose.FLIP_TOP_BOTTOM)