def __getitem__(self, index): e = self.meta[index] image_path = os.path.join(self.root, self.image_folder, f'{e["name"]}.jpg') layout_path = os.path.join(self.root, self.layout_folder, f'{e["name"]}.png') image, layout = self.paired_transform(load_image(image_path), load_image(layout_path)) edge = torch.from_numpy(self.load_edge_map(index) / 255).float() return {'image': image, 'label': layout, 'edge': edge, 'type': e['type']}
def __getitem__(self, index): image_path, label_path = self.filenames[index] image, label = self.paired_transform(load_image(image_path), load_image(label_path)) return { 'image': image, 'label': label, 'path': os.path.basename(label_path) }
def __getitem__(self, index): image_path, label_path = self.filenames[index] image = load_image(image_path).convert('RGB') layout = Image.fromarray(sio.loadmat(label_path)['fields']) image = F.resize(image, self.target_size, interpolation=Image.BILINEAR) layout = F.resize(layout, self.target_size, interpolation=Image.NEAREST) image = self.image_transform(image) layout = np.array(layout) layout[layout == 6] = 0 layout = torch.from_numpy(layout - 1).long() return { 'image': image, 'label': layout, 'path': os.path.basename(image_path) }
def __getitem__(self, index): return load_image(self.filenames[index]).convert('RGB')