def BuildTraining(batch_size, img_size, fea_label_file, train_index, validata_index): BATCH_SIZE = batch_size IMG_SIZE = img_size train_set = DataPrepare.DefaultDataset(fea_label_file, load_index=train_index) train_loader = torch.utils.data.DataLoader(train_set, batch_size=BATCH_SIZE, shuffle=True) validate_set = DataPrepare.DefaultDataset(fea_label_file, load_index=validata_index) validate_loader = torch.utils.data.DataLoader(validate_set, batch_size=BATCH_SIZE, shuffle=True) print('Data load Success') return train_loader, validate_loader
def BuildTesting(batch_size, img_size, fea_label_file): BATCH_SIZE = batch_size test_set = DataPrepare.DefaultDataset(fea_label_file) test_loader = torch.utils.data.DataLoader(test_set, batch_size=BATCH_SIZE, shuffle=False) print('Data load Success') return test_loader