python-3.x - 为对抗样本添加中值过滤器

Question

在此函数中添加 FGSM 后，如何在示例中添加中值滤波器。

def fgsm_attack(image, epsilon, data_grad):
    sign_data_grad = data_grad.sign()
    perturbed_image = image + epsilon*sign_data_grad
    perturbed_image = torch.clamp(perturbed_image, 0, 1)
    return perturbed_image

我使用此代码。 https://pytorch.org/tutorials/beginner/fgsm_tutorial.html

标签： python-3.ximage-processingpytorch