2021年1月25日星期一

Images for CNN model -- dimension issue

I am using images of driving data (i.e., trajectory of individual users) to predict their store visits in a CNN model. My images are mostly black and white, 360 x 360 pixels. When I convert the images to numpy array, the dimensions are 360 x 360 x 4, instead of 3 for RGB. A

from PIL import Image  from numpy import asarray  # load the image  image = Image.open(os.path.join(fig_path, 'image.png'))    # convert image to numpy array  data = asarray(image)  print(type(data))  # summarize shape  print(data.shape)    # create Pillow image  image2 = Image.fromarray(data)  print(type(image2))    # summarize image details  print(image2.mode)    print(image2.size)

I then stack all the images needed for the CNN using a temp[] vector and appending to it one by one and using . But when I input this entire matrix as input_shape into a CNN model, it breaks. Following is the error message:

link to my image

Any idea what's happening with my dimensions?

https://stackoverflow.com/questions/65893099/images-for-cnn-model-dimension-issue January 26, 2021 at 06:08AM

没有评论:

发表评论