search

hwalsuklee / tensorflow-fast-style-transfer

A simple, concise tensorflow implementation of fast style transfer
Apache License 2.0
245 stars 70 forks source link

I have trouble about these two Examples help #6

Open francistwwwww opened 7 years ago

francistwwwww commented 7 years ago
  1. 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'
  2. python run_test.py --content content/female_knight.jpg --style_model models/wave.ckpt --output result.jpg
    • AttributeError: module 'tensorflow' has no attribute 'pack'
hwalsuklee commented 7 years ago

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

  1. tf.pack must be changed into tf.stack for tensorflow version over 1.0
  2. 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 )

francistwwwww commented 7 years ago

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 ^^

hwalsuklee commented 7 years ago

Hey! check the code :D