When i run this code in jupyter notebook, I want to show some pictures of my datasets:
and the output is:
(1) The original picture is RGB, but the output looks like gary-image:confused: Why？
(2) There is no loop in the output, but why every pitctures output nine times, and it seems output three times in every channel? Why?
Your image tensor is of shape [batch, color channel, w, h]. You then squashed the last three dimensions together, and then reshaped into [batch, w, h, color channel] so your last three dimensions are messed up. And that is why you are seeing 9x subimages.
I don’t know why you squash all last three dimensions together at
imgs=... line and at first line in
show_images. If you get rid of them, then a single np call in the loop
plt.imshow(np.transpose(img, (1, 2, 0))) will work.
Thank you, I get rid of imgs=… in show_images and not to reshape the tensor, then i got the right output. Thank you again~~