Steganografiya quyidagi sohalarda qo'llaniladi, lekin ular bilan cheklanmaydi

triggers = {}

WM_X = np.array(ownership['inputs'])
nb_blocks = min(self.nb_blocks, len(WM_X))

step = floor(len(WM_X) / nb_blocks)
# For each block
for block in range(0, nb_blocks):
# Generate the encryption key
key = Fernet.generate_key()
# Add the key to the key set
keys.append(key)
# Initiating the encryption object based on the key
f = Fernet(key)
# Encrypt / store
to_encrypt = WM_X[block * step:(block + 1) * step].tobytes()
encrypted_trigger = f.encrypt(to_encrypt)
triggers['block_' + str(block)] = encrypted_trigger
shape_x, shape_y = WM_X.shape
triggers['shape'] = (int(shape_x / nb_blocks), shape_y)
triggers['dtype'] = WM_X.dtype

encrypted_trigger = {'triggers': triggers, 'keys': keys}

return encrypted_trigger

def decrypt_trigger(self, triggers, block_id, key):
""" Decrypt trigger block

Args:
triggers (dict): Encrypted trigger information
block_id (int): Trigger block to decrypt
key (str): Decryption key

Returns:
clear_trigger (array): Decrypted trigger

"""
shape = triggers['shape']
dtype = triggers['dtype']
f = Fernet(key)
clear_trigger = f.decrypt(triggers['block_' + str(block_id)])
clear_trigger = np.frombuffer(clear_trigger, dtype=dtype)
return clear_trigger.reshape(shape)

import hashlib
import hmac
import random
import warnings
from math import floor

import bitstring
import numpy as np
import pyfiglet
import torch