search

marcellacornia / mlnet

A Deep Multi-Level Network for Saliency Prediction. ICPR 2016
MIT License
94 stars 37 forks source link

Loading model error #11

Closed ltnghia closed 6 years ago

ltnghia commented 6 years ago

Hi, when I load the model, this error always happens: Layer weight shape (3, 3, 640, 64) not compatible with provided weight shape (64, 3, 3, 3)

How can I fix it? Thanks

marcellacornia commented 6 years ago

Hi @ltnghia, thanks for downloading our code.

Please check your Keras and Theano versions. This code is compatible with Keras 1.1.0 using Theano as backend.

Your keras.json file should have the following format:

{
    "image_dim_ordering": "th", 
    "epsilon": 1e-07, 
    "floatx": "float32", 
    "backend": "theano"
}