music.py 文件源码

def load_wav_chunks(filenames):
    num_files = len(filenames)
    max_chunks = 100000
    all_chunks = np.zeros([max_chunks, chunk_size])
    total_chunks = 0
    for file_idx in range(num_files):
        filename = filenames[file_idx]
        print("[" + str(file_idx).zfill(3) + "]: " + filename)
        rate, data = wavfile.read(filename)

        data = np.sum(data, axis=1)        
        print(data.shape)

        data = data.astype(np.float32) * (1.0 / 32768.0)

        chunks = create_chunks(data)
        num_chunks = len(chunks)

        all_chunks[total_chunks:total_chunks+num_chunks] = np.array(chunks)
        total_chunks += num_chunks

    all_chunks = all_chunks[0:total_chunks]

    return all_chunks