-
Notifications
You must be signed in to change notification settings - Fork 652
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Convert to Tensorflow #45
Comments
keras is a wrapper for tensorflow, turn on tensorflow as backend for keras and start from there? |
The most noteworthy change is to switch the axis of concatenation to the last channel instead of the first. This is because Theano is channels_first but Tensorflow is channels_last. E.g. change 'concatenate([up5, conv3], axis=1)' to 'concatenate([up5, conv3], axis=4)'. Otherwise, things can generally stay the same. |
Could you please elaborate more on the changes needed to use TensorFlow as backend? ValueError: Error when checking input: expected input_1 to have shape (None, 64, 64, 32, 1) but got array with shape (1, 1, 64, 64, 32) |
I think it's from "data.py", function "add_data_to_storage". Any help would be appreciated. |
To use Tensorflow as a backend, you don't need to change any of the code. Simply follow the instructions listed in the Keras backend documentation. |
Tested with
Works well |
This issue has been automatically marked as stale because it has not had recent activity. It will be closed if no further activity occurs. Thank you for your contributions. If you have questions, feel free to join the Slack group or email me at davidgellis2@gmail.com. |
do you have any suggestions on how to convert this Keras model to a Tensorflow model?
The text was updated successfully, but these errors were encountered: