pytorch的data_loader

train_loader = DataLoader(trainset,
                          batch_size=batch_size,
                          shuffle=True,
                          num_workers=num_workers,
                          pin_memory=True,
                          drop_last=True)

首先 DataLoader 为pytorch 内部类，此时只需要指定trainset， batch_size， shuffle， num_workers, ...等

最需要说名的为trainset：

trainset = Dataset(data_file=train_list,
                   data_dir=train_dir,
                   transform_trn=composed_trn,
                   transform_val=composed_val)

其中，data_file必须为图片文件名列表， data_dir为图片根目录， transform_trn为数据预处理方法，通过pytorch自带了

transforms.Compose（）函数指定。

data_file的文件名列表如下图所示：