Method __init__

attack_utils.py:128–134

(self, attack_types, flag=True)

class Attacker(nn.Module):
    def __init__(self, attack_types, flag=True):
        super().__init__()
        for attack_type in filter(lambda x : x != 'JPEGCompress', attack_types):
            setattr(self, attack_type, globals()[f'{attack_type}Attack']())

        if 'JPEGCompress' in attack_types:
            self.JPEGCompress = JPEGCompressAttack(flag)

    def forward(self, image, attack_type):
        return getattr(self, attack_type)(image)