(self, lmdb_dir, mask_dir, dataset, split, mode, input_size,
word_length)
| 84 | |
| 85 | class RefDataset(Dataset): |
| 86 | def __init__(self, lmdb_dir, mask_dir, dataset, split, mode, input_size, |
| 87 | word_length): |
| 88 | super(RefDataset, self).__init__() |
| 89 | self.lmdb_dir = lmdb_dir |
| 90 | self.mask_dir = mask_dir |
| 91 | self.dataset = dataset |
| 92 | self.split = split |
| 93 | self.mode = mode |
| 94 | self.input_size = (input_size, input_size) |
| 95 | #self.mask_size = [13, 26, 52] |
| 96 | self.word_length = word_length |
| 97 | self.mean = torch.tensor([0.485, 0.456, 0.406]).reshape(3, 1, 1) |
| 98 | self.std = torch.tensor([0.229, 0.224, 0.225]).reshape(3, 1, 1) |
| 99 | self.length = info[dataset][split] |
| 100 | self.env = None |
| 101 | # self.coco_transforms = make_coco_transforms(mode, cautious=False) |
| 102 | |
| 103 | def _init_db(self): |
| 104 | self.env = lmdb.open(self.lmdb_dir, |
nothing calls this directly
no outgoing calls
no test coverage detected