We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
2 parents 5e34bd1 + 1344a17 commit 1493bd6Copy full SHA for 1493bd6
site/en/r2/tutorials/generative/style_transfer.ipynb
@@ -217,7 +217,7 @@
217
"def load_img(path_to_img):\n",
218
" max_dim = 512\n",
219
" img = tf.io.read_file(path_to_img)\n",
220
- " img = tf.image.decode_jpeg(img)\n",
+ " img = tf.image.decode_image(img, channels=3)\n",
221
" img = tf.image.convert_image_dtype(img, tf.float32)\n",
222
"\n",
223
" shape = tf.cast(tf.shape(img)[:-1], tf.float32)\n",
0 commit comments