Spaces:
Build error
Build error
File size: 840 Bytes
1865436 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 |
from torch.utils.data import Dataset
class Labelizer():
def __init__(self):
super().__init__()
self.labels = {'background': 0, 'signboard': 1}
self.inv_labels = {0: 'background', 1: 'signboard'}
def transform(self, label):
return self.labels[label]
def inverse_transform(self, ys):
return self.inv_labels(ys)
def num_classes(self):
return len(self.labels)
class PoIDataset(Dataset):
def __init__(self,
data,
transforms=None):
self.data = data
self.transforms = transforms
def __len__(self):
return 1
def __getitem__(self, idx):
image = self.data
target = {}
if self.transforms is not None:
image = self.transforms(image)
return image, target |