(self, lmdb_dir, mask_dir, dataset, split, mode, input_size,
word_length)
| 97 | |
| 98 | class RefDataset(Dataset): |
| 99 | def __init__(self, lmdb_dir, mask_dir, dataset, split, mode, input_size, |
| 100 | word_length): |
| 101 | super(RefDataset, self).__init__() |
| 102 | self.lmdb_dir = lmdb_dir |
| 103 | self.mask_dir = mask_dir |
| 104 | self.dataset = dataset |
| 105 | self.split = split |
| 106 | self.mode = mode |
| 107 | self.input_size = (input_size, input_size) |
| 108 | #self.mask_size = [13, 26, 52] |
| 109 | self.word_length = word_length |
| 110 | self.mean = torch.tensor([0.485, 0.456, 0.406]).reshape(3, 1, 1) |
| 111 | self.std = torch.tensor([0.229, 0.224, 0.225]).reshape(3, 1, 1) |
| 112 | self.length = info[dataset][split] |
| 113 | self.env = None |
| 114 | def _init_db(self): |
| 115 | self.env = lmdb.open(self.lmdb_dir, |
| 116 | subdir=os.path.isdir(self.lmdb_dir), |
nothing calls this directly
no outgoing calls
no test coverage detected