data loader

from torch.utils.data import DataLoader
batch_size = 16
data = {
    'train':
    WheatDataset(
    '/home/yons/data/kaggle/global-wheat-detection/global-wheat-detection-coco-format/instances_train2017.json',
    '/home/yons/data/kaggle/global-wheat-detection/raw/train',
    get_train_transforms()
    ),
    'val':
    WheatDataset(
    '/home/yons/data/kaggle/global-wheat-detection/global-wheat-detection-coco-format/instances_val2017.json',
    '/home/yons/data/kaggle/global-wheat-detection/raw/train',
    get_valid_transforms()
    )
}
dataloaders = {
    'train': DataLoader(data['train'], batch_size=batch_size, shuffle=True),
    'val': DataLoader(data['val'], batch_size=batch_size, shuffle=False),
}
test_iter = iter(dataloaders['train'])
inputs, targets = next(test_iter)
inputs.shape, targets.shape