(image_root, gt_root, edge_root, batchsize, trainsize, shuffle=True, num_workers=12, pin_memory=True)
| 659 | |
| 660 | # dataloader for training |
| 661 | def get_loader(image_root, gt_root, edge_root, batchsize, trainsize, shuffle=True, num_workers=12, pin_memory=True): |
| 662 | dataset = PolypObjDataset(image_root, gt_root, edge_root, trainsize) |
| 663 | data_loader = data.DataLoader(dataset=dataset, |
| 664 | batch_size=batchsize, |
| 665 | shuffle=shuffle, |
| 666 | num_workers=num_workers, |
| 667 | pin_memory=pin_memory) |
| 668 | return data_loader |
| 669 | |
| 670 | # dataloader for training with no edge components |
| 671 | def get_loader_noEdge(image_root, gt_root, batchsize, trainsize, shuffle=True, num_workers=12, pin_memory=True): |
nothing calls this directly
no test coverage detected