记录一个错误@模型调用过程中遇到的一个错误
先看错误结果
代码中加载已有的.h5模型文件(假设为model)。在自己定义的攻击方法里面调用model。使用model对样本做预测,发现所有的预测标签都为同一类标签。即,模型已失效。
代码:
sess = tf.Session()
K.set_session(sess)
#加载模型
model = load_model(model_file)
#加载数据
_, _, X_test, Y_test = get_data(args.dataset,args.snr)
X_test, Y_test = X_test[:1000], Y_test[:1000]
#使用fgsm攻击方法
X_adv = fgsm(sess,X_test, Y_test, model, e)
#fgsm的定义
def fgsm(sess,x, y, model,e):
‘’’
:param x: 原始样本
:param y: 原始标签
:param model: 模型
:param model_name: 模型名称
:return: 保存得到的对抗样本
‘’’
WIDTH = 8
HIGH = 16
CHANNEL = 2
MAX = 1
MIN = -1
adv = []
# define e & err_num
err_num = 0