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')