ValueError:无法为形状为((1,)''的张量'Placeholder_765:0'提供形状(1、1)的值

时间:2018-07-02 09:46:01

标签: tensorflow keras

我从零开始在包含100个类的自定义数据集上训练了一个InceptionV3。在Keras上初始化CNN模型。我现在正尝试使用Foolbox生成此模型的对抗示例,但是出现上述错误。我要去哪里错了?库(Foolbox)似乎对其他人来说很好用,我的模型正确地通过了图像分类过程,没有任何错误,但是包装模型生成了它。

from keras.models import load_model
from keras.applications.vgg16 import VGG16
import foolbox
from foolbox.models import KerasModel
from foolbox.attacks import LBFGSAttack
from foolbox.criteria import TargetClass
import numpy as np
import foolbox
keras.backend.set_learning_phase(0)

model=load_model('standard_inceptionV3.h5')
fmodel = foolbox.models.KerasModel(model, bounds=(0, 255))

from PIL import Image
img = Image.open('/home/shikhar/Downloads/suit.jpeg')
img = img.resize((224,224))
img = np.asarray(img)
img = img[:, :, :3]

lab=model.predict(np.expand_dims(img, axis=0))
label=np.argmax(lab,axis=1)

from foolbox.criteria import Misclassification, TargetClass
attack = foolbox.attacks.FGSM(model=fmodel)
adversarial = attack(img, label,unpack=False)

0 个答案:

没有答案