Skip to content

Commit

Permalink
fix style image scale (#445)
Browse files Browse the repository at this point in the history
  • Loading branch information
iansimon committed Dec 9, 2016
1 parent e3e9b96 commit 4c5ebd4
Showing 1 changed file with 1 addition and 1 deletion.
Expand Up @@ -81,7 +81,7 @@ def main(unused_argv):

with tf.Graph().as_default():
style_end_points = learning.precompute_gram_matrices(
tf.expand_dims(tf.to_float(style_image) / 255.0, 0),
tf.expand_dims(tf.to_float(style_image), 0),
# We use 'pool5' instead of 'fc8' because a) fully-connected layers
# are already too deep in the network to be useful for style and b)
# they're quite expensive to store.
Expand Down

0 comments on commit 4c5ebd4

Please sign in to comment.