(self, path_img, path_gt, loadSize, mode=1)
| 23 | |
| 24 | class DocData(Dataset): |
| 25 | def __init__(self, path_img, path_gt, loadSize, mode=1): |
| 26 | super().__init__() |
| 27 | self.path_gt = path_gt |
| 28 | self.path_img = path_img |
| 29 | self.data_gt = os.listdir(path_gt) |
| 30 | self.data_img = os.listdir(path_img) |
| 31 | self.mode = mode |
| 32 | if mode == 1: |
| 33 | self.ImgTrans = (ImageTransform(loadSize)["train"], ImageTransform(loadSize)["train_gt"]) |
| 34 | else: |
| 35 | self.ImgTrans = ImageTransform(loadSize)["test"] |
| 36 | |
| 37 | def __len__(self): |
| 38 | return len(self.data_gt) |
nothing calls this directly
no test coverage detected