(image_root, gt_root, batchsize, trainsize, shuffle=False, num_workers=4, pin_memory=True, augmentation=False)
| 126 | |
| 127 | |
| 128 | def get_loader(image_root, gt_root, batchsize, trainsize, shuffle=False, num_workers=4, pin_memory=True, augmentation=False): #shuffle=True |
| 129 | |
| 130 | dataset = PolypDataset(image_root, gt_root, trainsize, augmentation) |
| 131 | data_loader = data.DataLoader(dataset=dataset, |
| 132 | batch_size=batchsize, |
| 133 | shuffle=shuffle, |
| 134 | num_workers=num_workers, |
| 135 | pin_memory=pin_memory) |
| 136 | return data_loader |
| 137 | |
| 138 | |
| 139 | class test_dataset: |
nothing calls this directly
no test coverage detected