Save_HCQT.py

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
Created on Fri Nov  9 12:53:00 2018

@author: wuyiming
"""

import numpy as np
from librosa.core import cqt,load,note_to_hz
from librosa.util import find_files
import const as C
import os

audiolist = find_files("Datas/audios_train")

path_hcqt = "Datas/specs_train"
itemcnt = len(audiolist)
i = 0
for audiofile in audiolist:
    i += 1
    print("Processing %d/%d" % (i,itemcnt))    
    wav,sr = load(audiofile,sr=C.SR)
    fmin = note_to_hz("C1")
    spec = np.stack([np.abs(cqt(wav,sr=C.SR,hop_length=C.H,n_bins=C.BIN_CNT,bins_per_octave=C.OCT_BIN,fmin=fmin*(h+1),filter_scale=2,tuning=None)).T.astype(np.float32) for h in range(C.CQT_H)])
    
    
    filename = audiofile.split('/')[-1].split(".")[0]
    
    savepath = os.path.join(path_hcqt,filename+".npy")
    np.save(savepath,spec)