diff --git a/outpaint.ipynb b/outpaint.ipynb index 68c0642..14dded2 100644 --- a/outpaint.ipynb +++ b/outpaint.ipynb @@ -537,12 +537,20 @@ "images = data.get_data(1)\n", "\n", "for i, image in enumerate(images):\n", + " # Show original picture\n", + " b, g, r = PIL.Image.fromarray(image).split()\n", + " IPython.display.display(PIL.Image.merge(\"RGB\", (r, g, b)))\n", + " \n", " image = image / 127.5 - 1\n", " image = recursive_paint(image)\n", " image = (image + 1) * 127.5\n", " image = image.astype(np.uint8)\n", " path = 'recursive/'+str(i)+'.jpg'\n", - " IPython.display.display(PIL.Image.fromarray(image))" + " \n", + " # Show after picture\n", + " b, g, r = PIL.Image.fromarray(image).split()\n", + " #PIL.Image.merge(\"RGB\", (r, g, b)).show() # This will show the pic in a \n", + " IPython.display.display(PIL.Image.merge(\"RGB\", (r, g, b)))" ] } ],