tensorflow-fast-style-transfer
tensorflow-fast-style-transfer copied to clipboard
I have trouble about these two Examples help
- python run_train.py --style style/wave.jpg --output model --trainDB train2014 --vgg_model pre_trained_model
- AttributeError: module 'tensorflow' has no attribute 'batch_matmul'
- python run_test.py --content content/female_knight.jpg --style_model models/wave.ckpt --output result.jpg
- AttributeError: module 'tensorflow' has no attribute 'pack'
Those are caused by tensorflow version mismatch. I think you're using tensorflow version over 1.0
Please check https://www.tensorflow.org/install/migration
- tf.pack must be changed into tf.stack for tensorflow version over 1.0
- tf.batch_matmul must be changed into tf.matmul for tensorflow version over 1.0
Or wait for a while until I update codes to be compatible to tensorflow version over 1.0 (I plan to update codes today :D )
Thank you hwalsuklee, You are so nice. Yes, I just pip install tensorflow, and the version should be over 1.0. Waiting for your updated code and thank you all for your contribution again ^^
Hey! check the code :D