search

flyyufelix / cnn_finetune

Fine-tune CNN in Keras
https://flyyufelix.github.io/2016/10/08/fine-tuning-in-keras-part2.html
MIT License
937 stars 413 forks source link

Image input shape issue #25

Closed Rabia-Metis closed 6 years ago

Rabia-Metis commented 6 years ago

I am using resnet-152 architecture and training input of size 224x224 and 331 images with tensor-flow backend but keep getting this error: ValueError: Error when checking input: expected data to have shape (None, 224, 224, 3) but got array with shape (331, 3, 224, 224) I tried setting image data format to channel_first and channel_last but its not working either way Kindly help me out

hiroshiperera commented 6 years ago

Hi were you abe to find an answer for this