search

marcellacornia / sam

Predicting Human Eye Fixations via an LSTM-based Saliency Attentive Model. IEEE Transactions on Image Processing (2018)
https://ieeexplore.ieee.org/document/8400593
MIT License
205 stars 76 forks source link

I met this error #26

Closed bemoregt closed 5 years ago

bemoregt commented 5 years ago

Hi, @baraldilorenzo @marcellacornia

I met this error:

ValueError: Dimension 2 in both shapes must be equal, but are 320 and 64. Shapes are [7,7,320,64] and [7,7,64,3]. for 'Assign' (op: 'Assign') with input shapes: [7,7,320,64], [7,7,64,3].

What's wrong to me?

marcellacornia commented 5 years ago

Have you installed the Keras and Theano versions specified in the README? Please also check your keras.json file and be sure to have "image_dim_ordering": "th" and "backend": "theano".