diff --git a/src/utils/preprocessor.py b/src/utils/preprocessor.py index 1b0d920..cb37c3e 100644 --- a/src/utils/preprocessor.py +++ b/src/utils/preprocessor.py @@ -1,5 +1,5 @@ import numpy as np -from scipy.misc import imread, imresize +from cv2 import imread, resize def preprocess_input(x, v2=True): @@ -16,7 +16,7 @@ def _imread(image_name): def _imresize(image_array, size): - return imresize(image_array, size) + return resize(image_array, size) def to_categorical(integer_classes, num_classes=2):