Method __init__

attack_utils.py:114–115

(self)

# these two functions are implemented based on WatermarkAttacker (https://github.com/XuandongZhao/WatermarkAttacker).
class VAEAttack(nn.Module):
    def __init__(self):
        super().__init__()
    
    def forward(self, image):
        out = self.model(image[None])