How to change the input image size?
Timo-hab opened this issue · comments
Hi,
if I change the image size of the input images, I get:
color_image = dataset.palette[prediction.ravel()].reshape(image_size)
ValueError: total size of new array must be unchanged
My input image has a resolution of 800x600x3 = 1440000
The length of color_image after "color_image = dataset.palette[prediction.ravel()]" is 1024x1024=1048576.
1048576 != 1440000. Thats the reason for the error, but how to fix it? Thank you in advance!
Ok, just change that line
if dataset.zoom > 1:
to
if dataset.zoom == 1: