pytorch加载语音数据文件,并进行批量处理
pytorch加载本地语音数据
Dataset
class Mydata(Dataset):
def __init__(self, root, train=True, transform=None, target_transform=None):
self.root = root
self.train = train
self.transform = transform
self.target_transform = target_transform
file_list_label = os.listdir(self.root)
self.label = []
self.data = []
for index, i in enumerate(file_list_label):
file_list_img = os.listdir(self.root + '/' + i)