Steganografiya quyidagi sohalarda qo'llaniladi, lekin ular bilan cheklanmaydi

loaders = self.generate_trigger_merrer()
elif self.args.trigger_technique == 'dawn':
loaders = self.generate_trigger_dawn()
else:
raise NotImplementedError

return loaders

def generate_trigger_noise(self):
"""Generation random trigger set.

Returns:
trainloader (Object): training loader
valloader (Object): validation loader
testloader (Object): test loader
triggerloader (Object): trigger loader
"""
batch_x, _ = self.trainset[0]
shapes = list(batch_x.shape)

# Compute the triggers
WM_X = torch.randn([self.args.trigger_size] + shapes)
labels = [int(i) for i in range(self.args.nbr_classes)]
WM_y = random.choices(labels, k=self.args.trigger_size)

watermarked_dataset, triggerset = [], []
for x, y in self.trainset:
watermarked_dataset.append((x, y))
for x, y in zip(WM_X, WM_y):
triggerset.append((x, y))
watermarked_dataset.append((x, y))

trainloader, valloader, testloader = self.loaders(watermarked_dataset)
triggerloader = torch.utils.data.DataLoader(
triggerset, batch_size=self.args.trigger_size, shuffle=True)

return trainloader, valloader, testloader, triggerloader

def generate_trigger_selected(self):
"""Generation trigger set with specific data.

Returns:
trainloader (Object): training loader
valloader (Object): validation loader
testloader (Object): test loader
triggerloader (Object): trigger loader
"""
watermarked_dataset = []
for x, y in self.trainset:
watermarked_dataset.append((x, y))
for x, y in self.specialset: