(image_root, gt_root, batchsize, trainsize, shuffle=True, num_workers=4, pin_memory=True)
| 71 | |
| 72 | |
| 73 | def get_loader(image_root, gt_root, batchsize, trainsize, shuffle=True, num_workers=4, pin_memory=True): |
| 74 | |
| 75 | dataset = PolypDataset(image_root, gt_root, trainsize) |
| 76 | data_loader = data.DataLoader(dataset=dataset, |
| 77 | batch_size=batchsize, |
| 78 | shuffle=shuffle, |
| 79 | num_workers=num_workers, |
| 80 | pin_memory=pin_memory) |
| 81 | return data_loader |
| 82 | |
| 83 | |
| 84 | class test_dataset: |
nothing calls this directly
no test coverage detected