uhfband / keras2caffe

Keras to Caffe model converter tool
MIT License
67 stars 22 forks source link

Support input_shape argument to the first layer #8

Closed abars closed 6 years ago

abars commented 6 years ago

Fix the problem occured when convert Sequential model like this. https://github.com/oarriaga/face_classification/blob/master/trained_models/gender_models/simple_CNN.81-0.96.hdf5

uhfband commented 6 years ago

Thank you!