search

LTS4 / DeepFool

A simple and accurate method to fool deep neural networks
355 stars 131 forks source link

Issue with test_deepfool.py #9

Open sumitb-123 opened 3 years ago

sumitb-123 commented 3 years ago

Instead of these lines tf = transforms.Compose([transforms.Normalize(mean=[0, 0, 0], std=map(lambda x: 1 / x, std)), transforms.Normalize(mean=map(lambda x: -x, mean), std=[1, 1, 1]), transforms.Lambda(clip), transforms.ToPILImage(), transforms.CenterCrop(224)])

it should be like

tf = transforms.Compose([transforms.Normalize(mean=[0, 0, 0], std=list(map(lambda x: 1 / x, std))), transforms.Normalize(mean=list(map(lambda x: -x, mean)), std=[1, 1, 1]), transforms.Lambda(clip), transforms.ToPILImage(), transforms.CenterCrop(224)]) i