pytorch加载不同size的文件,然后进行padding
以加载不同size的"XXX.npy"文件为例;
第一步:重写dataset,代码如下
from torch.utils.data import DataLoader, Dataset
class train_dataset(Dataset):
def __init__(self, train_path):
super(train_dataset, self)
self.all_list = find_files(train_path, ext="npy")
self.length = len(self.all_list)
def __getitem__(self, index):
x = np.load(self.all_list[index])
npy_name = self.all_list[index]
npys = "Ses" + npy_name