I'm trying to preprocess images dataset, represented in numpy array with images of shape (28, 28)
by rescaling them to (10, 10)
. I wrote a function for that:
import cv2 as cv
def resize_dataset(images):
resized_images = []
for img in images:
img = img.reshape((28,28))
resized_img = cv.resize(img, dsize=(10, 10))
resized_images.append(resized_img)
return numpy.array(resized_images)
But when I actually try to rescale them, I get the following error in cv.resize
:
error: OpenCV(4.0.0) /io/opencv/modules/imgproc/src/resize.cpp:3662: error: (-215:Assertion failed) func != 0 in function 'resize'
In google I only found people with the same error writing on c++ doing very different stuff, like this one: resize an image and changing its depth and this: http://answers.opencv.org/question/19715/error-215-func-0-in-function-convertto/
How do I fix it?
Oh, I actually figured it out. Images in the dataset were of type np.int64
. I just had to convert images to float32
, like this:
import numpy as np
import cv2 as cv
def resize_dataset(images):
resized_images = []
for img in images:
img = img.reshape((28,28)).astype('float32') # <-- convert image to float32
resized_img = cv.resize(img, dsize=(10, 10))
resized_images.append(resized_img)
return numpy.array(resized_images)
And now it works nicely. It looks like cv.resize
can't work with images represented in int. Hope this will help anyone