Function get_loader_scribble_noEdge_3326

utils/data_val.py:701–709

(image_root, gt_root, scribble_root, imgname_root, batchsize, trainsize, shuffle=True, num_workers=12, pin_memory=True)

    return data_loader

def get_loader_scribble_noEdge_3326(image_root, gt_root, scribble_root, imgname_root, batchsize, trainsize, shuffle=True, num_workers=12, pin_memory=True):
    dataset = PolypObjDataset_scribble_noEdge_3326(image_root, gt_root, scribble_root, imgname_root, trainsize)
    print('{}-{}-{}'.format(image_root,gt_root,imgname_root))
    data_loader = data.DataLoader(dataset=dataset,
                                  batch_size=batchsize,
                                  shuffle=shuffle,
                                  num_workers=num_workers,
                                  pin_memory=pin_memory)
    return data_loader

# test dataset and loader
class test_dataset: