-
Notifications
You must be signed in to change notification settings - Fork 4
/
__init__.py
executable file
·30 lines (26 loc) · 784 Bytes
/
__init__.py
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
from .oxford_pets import OxfordPets
from .eurosat import EuroSAT
from .ucf101 import UCF101
from .sun397 import SUN397
from .caltech101 import Caltech101
from .dtd import DescribableTextures
from .fgvc import FGVCAircraft
from .food101 import Food101
from .oxford_flowers import OxfordFlowers
from .stanford_cars import StanfordCars
from .fewsol import FewSOL
dataset_list = {
"oxford_pets": OxfordPets,
"eurosat": EuroSAT,
"ucf101": UCF101,
"sun397": SUN397,
"caltech101": Caltech101,
"dtd": DescribableTextures,
"fgvc": FGVCAircraft,
"food101": Food101,
"oxford_flowers": OxfordFlowers,
"stanford_cars": StanfordCars,
"fewsol": FewSOL
}
def build_dataset(dataset, root_path, shots):
return dataset_list[dataset](root_path, shots)