Invalid shape (28, 28, 1) for image data

Deep Learning with PyTorch: Zero to GANs Lecture 3 - Training Deep Neural Networks on a GPU
#1

lession-3_error
lession-3_error735×466 65.1 KB

Hello everyone,

I am getting invalid shape error when I ran this block of code,
can you please help me, what I did miss or any?

Thank you.

#2

squeeze() the image.

imshow(), when plotting grayscale image doesn’t need the channel.

1 Like
#3

It is working, Thank You.

But in lecture, Akash Sir has used permute method where it is working, so Why now it is showing error?

#4

Perhaps he used RGB (color) images. Or grey images that pretend to be RGB (information copied over the channels).

1 Like
#5

ya, It might be. Thank you.