kdllib.py 文件源码

def fetch_sample_audio_chords(n_samples=None):
    url = "https://dl.dropboxusercontent.com/u/15378192/piano_chords.tar.gz"
    partial_path = get_resource_dir("chords")
    full_path = os.path.join(partial_path, "piano_chords.tar.gz")
    if not os.path.exists(full_path):
        download(url, full_path)

    tf = tarfile.open(full_path)
    wav_names = [fname for fname in tf.getnames()
                 if ".wav" in fname.split(os.sep)[-1]]
    chords = []
    logger.info("Loading audio files...")
    for wav_name in wav_names[:n_samples]:
        f = tf.extractfile(wav_name)
        fs, d = wavfile.read(f)
        d = d.astype('float32') / (2 ** 15)
        chords.append(d)
    return fs, chords, wav_names