Steganografiya quyidagi sohalarda qo'llaniladi, lekin ular bilan cheklanmaydi

self.testloader, self.triggerloader) = loaders
else:
self.trainloader, self.valloader, self.testloader = self.loaders()

def get_model(self):
if self.args.trigger_technique == 'dawn':
return self.get_ownership(), DAWN(self.model, self.args)
else:
return self.model

def loaders(self, trainset=None):
"""Load the dataloaders for training.

Args:
trainset (Object, optional): trainset if watermark
activated

Returns:
trainloader (Object): training loader
valloader (Object): validation loader
testloader (Object): test loader

"""
if not trainset:
trainset = []
for x, y in self.trainset:
trainset.append((x, y))

trainloader = DataLoader(
trainset,
batch_size=self.args.batch_size,
shuffle=True)
testloader = DataLoader(
self.testset,
batch_size=self.args.batch_size,
shuffle=True)
valloader = DataLoader(
self.valset,
batch_size=self.args.batch_size,
shuffle=True)

return trainloader, valloader, testloader

def generate_trigger(self):
"""Generate trigger data for watermark.

Returns:
loaders (list): dataloaders with triggers
"""

if self.args.trigger_technique == 'noise':
loaders = self.generate_trigger_noise()
elif self.args.trigger_technique == 'selected':
loaders = self.generate_trigger_selected()
elif self.args.trigger_technique == 'patch':
loaders = self.generate_trigger_patch_msg()
elif self.args.trigger_technique == 'merrer':