使 Dataset 类迭代所有文件

问题描述

我有下面的数据集类，它从列表中检索一个项目并提取一些特征。
如何更改它以使其遍历文件列表中的每个项目并返回所有项目的特征，而不是仅返回索引的特征？

class MyDataset(Dataset):
    def __init__(self,data_dir,transform=None):    
        self.data_dir = data_dir
        self.file_lst = sorted([os.path.join(path,file) for path,dirs,files in os.walk('/folder/folder/folder') for file in files if fnmatch.fnmatch(file,'*wav')])
        self.transform = transforms.Compose(transform)
    
    def __len__(self):
        return len(self.file_lst)
    
    def __getitem__(self,idx):
        return self.transform(self.file_lst[idx])

train_dataset = MyDataset(input_dir,[file_reader,feat_extractor])
train_dataset[0]

解决方法

暂无找到可以解决该程序问题的有效方法，小编努力寻找整理中！

如果你已经找到好的解决方法，欢迎将解决方案带上本链接一起发送给小编。

小编邮箱:dio#foxmail.com (将#修改为@）