n_query=15,
**few_shot_params)
novel_loader = datamgr.get_data_loader(num_aug=params.gen_examples)
elif params.test_dataset == "EuroSAT":
print("Loading EuroSAT")
datamgr = EuroSAT_few_shot.SetDataManager2(image_size,
n_eposide=iter_num,
n_query=15,
**few_shot_params)
novel_loader = datamgr.get_data_loader(num_aug=params.gen_examples)
elif params.test_dataset == "CropDisease":
print("Loading CropDisease")
datamgr = CropDisease_few_shot.SetDataManager2(image_size,
n_eposide=iter_num,
n_query=15,
**few_shot_params)
novel_loader = datamgr.get_data_loader(num_aug=params.gen_examples)
elif params.test_dataset == "ChestX":
print("Loading ChestX")
datamgr = Chest_few_shot.SetDataManager2(image_size,
n_eposide=iter_num,
n_query=15,
**few_shot_params)
novel_loader = datamgr.get_data_loader(
num_aug=params.gen_examples) ### what if aug is true???
## uncomment code below to see if code is same across loaders
#########################################################################
outfile = os.path.join( checkpoint_dir.replace("checkpoints","features"), params.dataset + "_" + str(params.save_iter)+ ".hdf5")
else:
outfile = os.path.join( checkpoint_dir.replace("checkpoints","features"), params.dataset + ".hdf5")
if params.dataset in ["ISIC"]:
datamgr = ISIC_few_shot.SimpleDataManager(image_size, batch_size = 64)
data_loader = datamgr.get_data_loader(aug = False )
elif params.dataset in ["EuroSAT"]:
datamgr = EuroSAT_few_shot.SimpleDataManager(image_size, batch_size = 64)
data_loader = datamgr.get_data_loader(aug = False )
elif params.dataset in ["CropDisease"]:
datamgr = CropDisease_few_shot.SimpleDataManager(image_size, batch_size = 64)
data_loader = datamgr.get_data_loader(aug = False )
elif params.dataset in ["ChestX"]:
datamgr = Chest_few_shot.SimpleDataManager(image_size, batch_size = 64)
data_loader = datamgr.get_data_loader(aug = False )
model = model_dict[params.model]()
model = model.cuda()
tmp = torch.load(modelfile)
state = tmp['state']
state_keys = list(state.keys())
for i, key in enumerate(state_keys):