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的文件名列表如下圖所示：