def get_passport(passport_data, device):
n = 20 # any number
key_y, y_inds = passport_generator.get_key(passport_data, n)
key_y = key_y.to(device)
key_x, x_inds = passport_generator.get_key(passport_data, n)
key_x = key_x.to(device)
return key_x, key_y
def setup_keys(self, pretrained_model):
if self.key_type != 'random':
n = 1 if self.key_type == 'image' else 20 # any number will do
key_x, x_inds = passport_generator.get_key(self.passport_data, n)
key_x = key_x.to(self.device)
key_y, y_inds = passport_generator.get_key(self.passport_data, n)
key_y = key_y.to(self.device)
passport_generator.set_key(pretrained_model, self.model,
key_x, key_y)
