コード例 #1
0
    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
コード例 #2
0
    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)
コード例 #3
0
    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
コード例 #4
0
    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
コード例 #5
0
 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)