2017-08-28 26 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順序模型,並且此代碼能夠執行。爲什麼預訓練模型會有所不同? –