Python Adversarial_Szegedy.saveの例

コード例 #1

ファイル: test_adversarial_cifar100.py プロジェクト: mducoffe/Adv_detect

def test_Adversarial(option, repo, filename):
    if option=='Adversarial_Szegedy':
        adv_object = Adversarial_Szegedy(repo=repo, filename=filename, nb_class=100, data_name='cifar100')
        #adv_object.generate()
        adv_object.save('./adv/Szegedy', 'cifar100_v0')
    if option=='Adversarial_Goodfellow':
        adv_object = Adversarial_Goodfellow(repo=repo, filename=filename, nb_class=100, data_name='cifar100')
        adv_object.generate() 
        adv_object.save('./adv/Goodfellow', 'cifar100_v0')

コード例 #2

def test_Adversarial(option):
    if option == 'Adversarial_Szegedy':
        adv_object = Adversarial_Szegedy()
        #adv_object.generate()
        adv_object.save('./adv/Szegedy', 'cifar10_v0')
    if option == 'Adversarial_Goodfellow':
        adv_object = Adversarial_Goodfellow()
        #adv_object.generate()
        adv_object.save('./adv/Goodfellow', 'cifar10_v0')

    if option == 'Adversarial_DeepFool':
        adv_object = Adversarial_DeepFool()
        adv_object.generate()