utils.py 文件源码

def data_loader(image_root, data_list, shuffle=True, batch_size=64, workers=20, is_cuda=True, is_visualization=False):
    kwargs = {'num_workers': workers, 'pin_memory': True} if is_cuda else {}
    transform=transforms.Compose([
            trans.person_crop(ratio=(1, 0.75),crop_type=1),\
            trans.scale(size=(64, 128)),\
            transforms.ToTensor()
            ])
    preid = dataset.listDataset(
            image_root, 
            data_list,
            shuffle,
            transform=transform, 
            is_visualization=is_visualization)

    data_loader = torch.utils.data.DataLoader(preid,
            batch_size=batch_size, 
            shuffle=True, 
            **kwargs)

    return data_loader