Function
get_loader
(image_root, gt_root, batchsize, trainsize, shuffle=False, num_workers=4, pin_memory=True, augmentation=False, split='train', color_image=True)
Source from the content-addressed store, hash-verified
| 95 | return self.size |
| 96 | |
| 97 | def get_loader(image_root, gt_root, batchsize, trainsize, shuffle=False, num_workers=4, pin_memory=True, augmentation=False, split='train', color_image=True): |
| 98 | dataset = PolypDataset(image_root, gt_root, trainsize, augmentation, split, color_image) |
| 99 | return data.DataLoader(dataset=dataset, batch_size=batchsize, shuffle=shuffle, num_workers=num_workers, pin_memory=pin_memory) |
| 100 | |