def main():
    from torchvision import transforms
    from torch.utils import data

    tf_source = transforms.Compose([
        transforms.Resize(64),
        transforms.ToTensor(),
        transforms.Normalize(mean=(0.1307,), std=(0.3081,))
    ])



    dataset = KittiLoader('/home/abderrazzak/kitti',  transforms=tf_source)

    print(len(dataset))

    train_loader = data.DataLoader(dataset=dataset,
                                    batch_size=1, shuffle=True,)
    print(len(train_loader))

    for x, y in train_loader:
       print(x.size())
       print(y)
       break

if __name__ == '__main__':
    main()