File size: 589 Bytes
57746f1 |
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 |
import copy
import itertools
import functools
import numpy as np
import torch
import torch.utils.data
import torchvision.transforms as torch_transforms
import encoding.datasets as enc_ds
encoding_datasets = {
x: functools.partial(enc_ds.get_dataset, x)
for x in ["coco", "ade20k", "pascal_voc", "pascal_aug", "pcontext", "citys"]
}
def get_dataset(name, **kwargs):
if name in encoding_datasets:
return encoding_datasets[name.lower()](**kwargs)
assert False, f"dataset {name} not found"
def get_available_datasets():
return list(encoding_datasets.keys())
|