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:
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
没有评论:
发表评论