MusicGen/spectogram_creator.py at master · ParshwaS/MusicGen · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
from riffusion_helper.spectrogram_image_converter import SpectrogramImageConverter
from riffusion_helper.spectrogram_params import SpectrogramParams
import os
import pydub
from glob import glob

params = SpectrogramParams()

converter = SpectrogramImageConverter(params=params)

def save_spectrogram_image_from_audio(filename: str):

    audio = pydub.AudioSegment.from_file(f'{filename}')

    print(len(audio))

    name = filename.split('/')[-1].split('.')[0]

    for i in range(0, len(audio), 5000):
        clip = audio[i:i+5000]
        image = converter.spectrogram_image_from_audio(clip)
        image.save(f'./dataset/spectrograms/{name}_{i}.png')
        print(f'./dataset/spectrograms/{name}_{i}.png')

    return True

# Load all files with .mid from dataset folder

# datadir = './dataset/lpd_5_cleansed'

# files = glob(f'{datadir}/A/A/A/*/*.mid')

# print(files)

# for file in files:
save_spectrogram_image_from_audio("./output.mp3")
    # break

print('Done')