2017-08-28 109 views
0

代码:Keras InceptionV3 model.predict

from keras.applications import InceptionV3 
    model = InceptionV3(weights="imagenet") 
    shape = (None,image_size,image_size,num_channels) 
    x = tf.placeholder(tf.float32, shape=shape)  
    adv_x,grad_x = fgm(x, model, model.predict(x), y=y, targeted=True, eps=0, clip_min=-0.5, clip_max=0.5) 
    adv_,grad_ = batch_eval(sess, [x,y], [adv_x,grad_x], [inputs,targets], args={'batch_size': args['batch_size']}) 

    model.predict(x) 

错误:

File "/u/.../env/lib/python3.5/site-packages/keras/engine/training.py", line 1594, in predict 
    batch_size=batch_size, verbose=verbose) 
    File "/u/.../env/lib/python3.5/site-packages/keras/engine/training.py", line 1208, in _predict_loop 
    batches = _make_batches(samples, batch_size) 
    File "/u/.../env/lib/python3.5/site-packages/keras/engine/training.py", line 364, in _make_batches 
    num_batches = int(np.ceil(size/float(batch_size))) 
TypeError: unsupported operand type(s) for /: 'Dimension' and 'float' 

我可以在实际图像使用model.predict,但对tf.placeholders或tf.variables这个错误结束 任何人都可以帮助我调试这个错误?

回答

0

Keras'Model.predict需要一个numpy数组作为输入数据。您可能还需要包括batch_size值,除非您的批量大小为32从文档:

predict(self, x, batch_size=None, verbose=0, steps=None) method of keras.engine.training.Model instance 
    Generates output predictions for the input samples. 

    Computation is done in batches. 

    # Arguments 
     x: The input data, as a Numpy array 
      (or list of Numpy arrays if the model has multiple outputs). 
     batch_size: Integer. If unspecified, it will default to 32. 
     verbose: Verbosity mode, 0 or 1. 
     steps: Total number of steps (batches of samples) 
      before declaring the prediction round finished. 
      Ignored with the default value of `None`. 

    # Returns 
     Numpy array(s) of predictions. 

    # Raises 
     ValueError: In case of mismatch between the provided 
      input data and the model's expectations, 
+0

我使用Cleverhans图书馆,象征性的攻击是建立在tensorflow。编辑上面的问题。我应该能够做到这一点,因为我建立了自己的MNIST和CIFAR顺序模型,并且此代码能够执行。为什么预训练模型会有所不同? –