24 lines
813 B
Python
24 lines
813 B
Python
import os
|
|
|
|
from .utils import Datum, DatasetBase
|
|
from .oxford_pets import OxfordPets
|
|
|
|
|
|
template = ['a photo of a {}.']
|
|
|
|
|
|
class Caltech101_TSNE(DatasetBase):
|
|
|
|
dataset_dir = 'caltech-101_tsne'
|
|
|
|
def __init__(self, root, num_shots):
|
|
self.dataset_dir = os.path.join(root, self.dataset_dir)
|
|
self.image_dir = os.path.join(self.dataset_dir, '101_ObjectCategories')
|
|
self.split_path = os.path.join(self.dataset_dir, 'split_zhou_Caltech101.json')
|
|
|
|
self.template = template
|
|
|
|
train, val, test = OxfordPets.read_split(self.split_path, self.image_dir)
|
|
train = self.generate_fewshot_dataset(train, num_shots=num_shots)
|
|
t_sne_test = self.generate_fewshot_dataset(test, num_shots=num_shots)
|
|
super().__init__(train_x=train, val=val, test=test,t_sne=t_sne_test) |