(image_root, gt_root, scribble_root, batchsize, trainsize, shuffle=True, num_workers=12, pin_memory=True)
| 690 | |
| 691 | # dataloader for training with scribble and no edge components |
| 692 | def get_loader_scribble_noEdge(image_root, gt_root, scribble_root, batchsize, trainsize, shuffle=True, num_workers=12, pin_memory=True): |
| 693 | dataset = PolypObjDataset_scribble_noEdge(image_root, gt_root, scribble_root, trainsize) |
| 694 | data_loader = data.DataLoader(dataset=dataset, |
| 695 | batch_size=batchsize, |
| 696 | shuffle=shuffle, |
| 697 | num_workers=num_workers, |
| 698 | pin_memory=pin_memory) |
| 699 | return data_loader |
| 700 | |
| 701 | def get_loader_scribble_noEdge_3326(image_root, gt_root, scribble_root, imgname_root, batchsize, trainsize, shuffle=True, num_workers=12, pin_memory=True): |
| 702 | dataset = PolypObjDataset_scribble_noEdge_3326(image_root, gt_root, scribble_root, imgname_root, trainsize) |
nothing calls this directly
no test coverage detected