def __getitem__(self, index): env = self.env with env.begin(write=False) as txn: bin_file = txn.get(self.keys[index]) data, target = loads(bin_file) if self.transform is not None: data = self.transform(data) if self.target_transform is not None: target = self.target_transform(target) return data, target
def __getitem__(self, index): sample = loads(self._txn.get(self.keys[index])) img = sample[0] label = sample[1] encoded_jpg_io = io.BytesIO(img) image = Image.open(encoded_jpg_io) if self.transform is not None: image = self.transform(image) if self.target_transform is not None: label = self.target_transform(label) return (image, label)
def __getitem__(self, index): env = self.env with env.begin(write=False) as txn: bin_file = txn.get(self.keys[index]) image, target = loads(bin_file) if self.is_image: image = self._image_decode(image) if self.transform is not None: image = self.transform(image) if self.target_transform is not None: target = self.target_transform(target) return image, target
def _set_keys(self): self.keys = self._txn.get(b'__keys__') self.keys = loads(self.keys) self.classes = self._txn.get(b'__classes__') self.classes = loads(self.classes)