生成音频训练文件

'''
生成训练集和测试集，保存在txt文件中
'''

import os
import random


train_ratio = 1


test_ratio = 1-train_ratio

rootdata = r"dataset/corpus/test"

train_list, test_list = [],[]
data_list = []

class_flag = 200
for a,b,c in os.walk(rootdata):
    print('a=', a)
    for j in c:
        biaoqian=os.path.basename(a)
        if os.path.splitext(j)[1] == '.wav':
            train_data = os.path.join(a, j) + '\t' + str(class_flag) + '\n'
            train_list.append(train_data)
    class_flag += 1
    #print(train_list)
    random.shuffle(train_list)
    with open('test.txt', 'a', encoding='UTF-8') as f:
        for train_img in train_list:
            f.write(str(train_img))
    train_list, test_list = [], []
    if class_flag==241:
        break